Commit Graph

24 Commits (93d5c1ed5afc83223d8ff2f58177f7a8a1219868)

Author SHA1 Message Date
Yu Yang 9bd70a1e04 Change tensor uses proto::VarType::type
6 years ago
qingqing01 731d45a39a
Enable BatchNorm to use global mean and variane during training (#14630)
6 years ago
Sylwester Fraczek 8a1eeec579 add mkldnn prop_kind phase for inference-only case to pooling and activations (#14278)
6 years ago
chengduo 6c6e638550
Add InferVarType for some op (#14201)
6 years ago
Xin Pan 8f2116d8fa clean up after the changes have been stopped for so long.
6 years ago
guochaorong b1fc238694 Revert "Disable in_place in batch_norm API. (#12736)"
7 years ago
qingqing01 f5d5d7b2d9
Disable in_place in batch_norm API. (#12736)
7 years ago
qingqing01 10fbb831ed
Skip BatchNorm when feature only has 1 element. (#11578)
7 years ago
pzelazko-intel 9a15c92317 bnorm+relu fuse for mkldnn (inference) (#11434)
7 years ago
Luo Tao 5aac910b89 add url of cuda9.0_cudnn7_avx_mkl library
7 years ago
tensor-tang 489f581d49
Merge pull request #11098 from mozga-intel/mozga-intel/Batch_norm_mkldnn_layout
7 years ago
dzhwinter bfa3fd6f15
add inplace attribute to op_proto_maker (#10665)
7 years ago
mozga-intel 7d5643562f MKLDNN layout: Support for batch norm operator
7 years ago
mozga-intel 3ff9ba0e6b Mkldnn layout (#11040)
7 years ago
Yu Yang 0e78cb69fb Clean OpProtoAndCheckerMaker
7 years ago
Tomasz Patejko 4a497b826d MKLDNN implementation of batch normalization (#9904)
7 years ago
dzhwinter 57be5c6c74
"fix double type error" (#10322)
7 years ago
Siddharth Goyal cea391217a
Fix cpplint errors (#9800)
7 years ago
Yu Yang 0760aaf440 Shrink batch_norm_grad's inputs
7 years ago
Kexin Zhao 446d54f5c3 update
7 years ago
Kexin Zhao e870947cfd fix batch norm fp16 param type
7 years ago
qingqing01 24509f4af9 Fix the grammar in copyright. (#8403)
7 years ago
Yi Wang fc374821dd Correct #include path
7 years ago
Yi Wang 90648f336d Move file to fluid/; Edit CMakeLists.txt
7 years ago