TF2 icon indicating copy to clipboard operation
TF2 copied to clipboard

batch nomalization

Open zhaohb opened this issue 4 years ago • 1 comments

Hi,

in resnet50 has batch nomalization layer, but I can not find bn kernel in device client, why?

thank you

zhaohb avatar Mar 10 '20 02:03 zhaohb

The Batch Normalization calculation locates in pe.cl. We recommend you to review the device code by following the data stream kernel by kernel, then you will find all the procedures. Or you can also have a look at the model header file like resnet50.h to build a thorough understanding of the whole structure of Runtime_Engine. Because all the varaible names in the header file are meanful, you can first make a guess of their functions and then trace them back to the kernel code to explore their real behavior.

Abraham-Xu avatar Mar 12 '20 10:03 Abraham-Xu