From 02d186814f580e2ea0545e908b04e27fe1ad4c21 Mon Sep 17 00:00:00 2001 From: Ian Wong Date: Tue, 5 Dec 2017 15:07:59 +0800 Subject: [PATCH] Support NHWC for LRN and batch normalization, and optimize batch normalization speed (#1936) * 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