Home
last modified time | relevance | path

Searched refs:abstract_scalar (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcombine_cast_fusion.cc124 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in Run() local
126 MS_EXCEPTION_IF_NULL(abstract_scalar); in Run()
127 index->set_abstract(abstract_scalar); in Run()
Dconcat_outputs_for_all_gather.cc152 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in Process() local
153 idx->set_abstract(abstract_scalar); in Process()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Doperator_info.cc120 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(index); in CreateMultipleOutputsOfAnfNode() local
121 MS_CHECK_TRUE_RET(abstract_scalar != nullptr, lite::RET_ERROR); in CreateMultipleOutputsOfAnfNode()
122 idx->set_abstract(abstract_scalar); in CreateMultipleOutputsOfAnfNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dsplit_fission.cc75 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in CreateTupleGetItem() local
76 idx->set_abstract(abstract_scalar); in CreateTupleGetItem()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dconcat_outputs_for_all_gather.cc159 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in Process() local
160 idx->set_abstract(abstract_scalar); in Process()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc307 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in GetMultipleOutputsOfAnfNode() local
308 MS_CHECK_TRUE_MSG(abstract_scalar != nullptr, false, "create AbstractScalar return nullptr"); in GetMultipleOutputsOfAnfNode()
309 index->set_abstract(abstract_scalar); in GetMultipleOutputsOfAnfNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dcommunication_op_fusion.cc424 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(); in DoFusion() local
425 MS_EXCEPTION_IF_NULL(abstract_scalar); in DoFusion()
426 index->set_abstract(abstract_scalar); in DoFusion()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc183 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in CreateMultipleOutputsOfAnfNode() local
184 idx->set_abstract(abstract_scalar); in CreateMultipleOutputsOfAnfNode()
491 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in CreatTupleGetItemNode() local
492 idx->set_abstract(abstract_scalar); in CreatTupleGetItemNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc154 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in CreateTupleGetItem() local
155 idx->set_abstract(abstract_scalar); in CreateTupleGetItem()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc345 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in CreateTupleGetItem() local
346 idx->set_abstract(abstract_scalar); in CreateTupleGetItem()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.cc669 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in CreatTupleGetItemNode() local
670 idx->set_abstract(abstract_scalar); in CreatTupleGetItemNode()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc627 auto abstract_scalar = std::make_shared<abstract::AbstractScalar>(imm); in CreatTupleGetItemNode() local
628 idx->set_abstract(abstract_scalar); in CreatTupleGetItemNode()