Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dmomentum_lossscale_fusion.cc64 CheckCNodeInputSize(mul_cnode, kMulInputTensorNum); in Process()
67 for (size_t i = 1; i < kMulInputTensorNum + 1; ++i) { in Process()
86 … mul_cnode->input(kMulInputTensorNum + 1 - value_node_index), in Process()
Dmul_addn_fusion.cc35 inputs.push_back(mul->input(kMulInputTensorNum + 1 - lossscale_input_index)); in CreateFusionNode()
76 if (mul == nullptr || AnfAlgo::GetInputTensorNum(mul) != kMulInputTensorNum) { in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc257 CheckCNodeInputSize(mul_node, kMulInputTensorNum); in CreateTile()
341 CheckCNodeInputSize(mul_node, kMulInputTensorNum); in GetDependNode()
439 CheckCNodeInputSize(mul_node, kMulInputTensorNum); in Process()
556 CheckCNodeInputSize(mul_node, kMulInputTensorNum); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.h43 constexpr size_t kMulInputTensorNum = 2; variable