Home
last modified time | relevance | path

Searched refs:kCastInputTensorNum (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dlayer_norm_beta_gamma_backprop_fusion.cc149 CheckCNodeInputSize(cast_nodes[0], kCastInputTensorNum); in Process()
150 CheckCNodeInputSize(cast_nodes[1], kCastInputTensorNum); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dmerge_cast_to_op.cc182 if (AnfAlgo::GetInputTensorNum(node) < kCastInputTensorNum) { in MergeCastToNextOp()
184 << ", should be not less than " << kCastInputTensorNum; in MergeCastToNextOp()
220 CheckCNodeInputSize(cur_node, kCastInputTensorNum); in MergeCastToPriorOp()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.h33 constexpr size_t kCastInputTensorNum = 1; variable