nnvm/tests/python/unittest · 4bd92a4a2123f0d70b6de34bd995878ac1d57a26 · wenyuanbo / tic

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

committed May 29, 2018

2374852e

Name	Last commit	Last update
..
test_graph.py		Loading commit data...
test_infer_shape.py		Loading commit data...
test_symbol.py		Loading commit data...
test_top_level1.py		Loading commit data...
test_top_level2.py		Loading commit data...
test_top_level3.py		Loading commit data...
test_top_level4.py		Loading commit data...