Searched defs:batchnorm (Results 1 – 2 of 2) sorted by relevance
29 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() local120 CNodePtr batchnorm = nullptr; in Process() local
844 auto batchnorm = static_cast<kernel::BatchnormCPUKernel *>(kernel->kernel()); in AdminSetupVirtualBatch() local