Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dbatchnorm_to_bninfer.cc29 CNodePtr CreateBNInfer(const FuncGraphPtr &graph, const CNodePtr &batchnorm, const AnfNodePtr &node… in CreateBNInfer()
62 bool CheckBatchNorm(const FuncGraphPtr &graph, const CNodePtr &batchnorm) { in CheckBatchNorm()
85 bool NeedFusion(const FuncGraphPtr &graph, const AnfNodePtr &node, CNodePtr *batchnorm) { in NeedFusion()
111 VectorRef batchnorm({prim::kPrimBatchNorm, Xs}); in DefinePattern() local
120 CNodePtr batchnorm = nullptr; in Process() local
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_session.cc844 auto batchnorm = static_cast<kernel::BatchnormCPUKernel *>(kernel->kernel()); in AdminSetupVirtualBatch() local