• Pruthvi's avatar
    Pruthvi/bn inference (#670) · 5394ad2d
    Pruthvi authored
    * Added new ctor for bn which supports Inference
    - added mkldnn emitter code for bn inference
    * Added test case for bn inference
    - added support for layout propogation for bn inference
    * added sanity checks for gamma, beta, mean, variance shape in bn
    * added serializer support for bn inference
    5394ad2d
batch_norm.hpp 3.18 KB