Home
last modified time | relevance | path

Searched refs:kTupleGetItemInputTensorNum (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dgetitem_tuple.cc48 CheckCNodeInputSize(tuple_getitem, kTupleGetItemInputTensorNum); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dbatchnorm_to_bninfer.cc90 CheckCNodeInputSize(tuple_getitem, kTupleGetItemInputTensorNum); in NeedFusion()
Dbatchnormgrad_to_bninfergrad.cc92 CheckCNodeInputSize(tuple_getitem, kTupleGetItemInputTensorNum); in NeedFusion()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.h38 constexpr size_t kTupleGetItemInputTensorNum = 2; variable
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dmerge_cast_to_op.cc196 CheckCNodeInputSize(x_cnode, kTupleGetItemInputTensorNum); in GetPriorOp()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Ddropout_unify_mindir.cc287 CheckCNodeInputSize(tuple_cnode, kTupleGetItemInputTensorNum); in Process()