Home
last modified time | relevance | path

Searched defs:batchnorm (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dbatchnorm_to_bninfer.cc61 bool CheckBatchNorm(const FuncGraphPtr &graph, const CNodePtr &batchnorm) { in CheckBatchNorm()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_session.cc1067 auto batchnorm = static_cast<kernel::LiteKernel *>(kernel->kernel()); in AdminSetupVirtualBatch() local
/third_party/mindspore/mindspore-src/source/third_party/proto/coreml/
DNeuralNetwork.proto485 BatchnormLayerParams batchnorm = 160; field