Skip to content

Commit ee998a9

Browse files
authored
CPU Batch Norm Op (#4964)
* init batch norm op * prepare input output * compute mean_out var_out save_mean save_var on CPU * active is test * use eigen to do computation * complete batch norm forward * set default momentum to 0.9 * add batch norm grad op in CPU * add tensor_format and NHWC support, add python test * add test training * add batch norm gradient test * improve comment, fix foward Python UnitTest * add gradient test * fix eigen warning * follow name style * fix a bug * change float to T * add simple forward test * test with different place * add backward test * refine python test * remove old python test code * code clean * follow code style * update comment
1 parent b54990e commit ee998a9

File tree

5 files changed

+663
-3
lines changed

5 files changed

+663
-3
lines changed

cmake/external/eigen.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ ExternalProject_Add(
88
extern_eigen3
99
${EXTERNAL_PROJECT_LOG_ARGS}
1010
GIT_REPOSITORY "https://github.com/RLovelett/eigen.git"
11-
GIT_TAG 4e79cb69b9425f5f8c3a84be4350d4ab75b5fd9d
11+
GIT_TAG 70661066beef694cadf6c304d0d07e0758825c10
1212
PREFIX ${EIGEN_SOURCE_DIR}
1313
UPDATE_COMMAND ""
1414
CONFIGURE_COMMAND ""

0 commit comments

Comments
 (0)