Skip to content
Snippets Groups Projects
Commit 2374852e authored by guopinglong's avatar guopinglong Committed by Tianqi Chen
Browse files

make shape inference of BatchNorm layout neutral (#301) 

* 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
parent 7112dd78
No related branches found
No related tags found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment