test_infer_shape.py
8.29 KB
-
make shape inference of BatchNorm layout neutral (#301) · 2374852e* make shape inference of BatchNorm layout neutral * refactor to use the axis variable to do BatchNorm shape inference * refactor to use the axis variable to do BatchNorm shape inference * add unittest to the axis param for batch norm shape inference
guopinglong committed