Support NHWC for LRN and batch normalization, and optimize batch normalization speed (#1936)

Browse source

* add LRN and LRNGrad

* add LRNGrad

* support bn with NHWC

* support bn with NHWC

* add batch norm grad

* support tf bnv2/bngradv2

* fix uts

* fix uts

* fix ut

* fix ut

* fix ut

* fix ut

* fix style issue

* fix ut

* fix ut

* fix ut

* update python api and docs

* meet code review

* meet code review

* fix ut

* fix ut

* fix ut

...

This commit is contained in:

Ian Wong 2017-12-05 15:07:59 +08:00 committed by GitHub

parent 4671809646

commit 02d186814f

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Show stats Download patch file Download diff file Expand all files Collapse all files

Diff content is not available