Searched refs:kBatchNormGradOpName (Results 1 – 5 of 5) sorted by relevance
38 …ctor<AnfNodePtr> bn_grad_inputs = {NewValueNode(std::make_shared<Primitive>(kBatchNormGradOpName)), in CreateNewBatchNormGrad()63 auto prim = std::make_shared<Primitive>(kBatchNormGradOpName); in DefinePattern()
52 {kBatchNormGradOpName, {{0, 1}, {0}}},
89 auto prim = std::make_shared<Primitive>(kBatchNormGradOpName); in DefinePattern()
155 constexpr auto kBatchNormGradOpName = "BatchNormGrad"; variable665 kBatchNorm, kBatchNormGradOpName};
120 if (kernel_name == kBatchNormGradOpName) { in Init()