TF2
TF2 copied to clipboard
batch nomalization
Hi,
in resnet50 has batch nomalization layer, but I can not find bn kernel in device client, why?
thank you
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.