Home
last modified time | relevance | path

Searched refs:kBNTrainingUpdateGradOpName (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbatch_norm_grad_split.cc38 NewValueNode(std::make_shared<Primitive>(kBNTrainingUpdateGradOpName)), bn_grad_inputs[kIndex1], in CreateOutputsOfUpdateGrad()
Dbn_grad_split.cc39 NewValueNode(std::make_shared<Primitive>(kBNTrainingUpdateGradOpName)), bn_grad_inputs[kIndex1], in CreateOutputsOfUpdateGrad()
Dbatch_norm_grad_infer_fission.cc125 NewValueNode(std::make_shared<Primitive>(kBNTrainingUpdateGradOpName)), in CreateBNTrainingUpdateGrad()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h129 constexpr auto kBNTrainingUpdateGradOpName = "BNTrainingUpdateGrad"; variable