Home
last modified time | relevance | path

Searched defs:axis_input (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dgather_fusion_mapper.cc37 auto axis_input = cnode->input(kNameGatherInputNum - 1); in Mapper() local
Dsqueeze_mapper.cc92 auto axis_input = cnode->input(axis_input_index); in Mapper() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/
Ddvm_kernel_build.cc48 ShapeVector GetAxisList(const AnfNodePtr &axis_input) { in GetAxisList()
436 dvm::ShapeRef *CacheAxis(const CNodePtr &node, const AnfNodePtr &axis_input) { in CacheAxis()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_generate_strategy.cc196 auto axis_input = GetValue<int64_t>(op->input_value().at(2)); in GetGatherAxis() local
1332 auto axis_input = GetValue<int64_t>(op->input_value().at(1)); in PrepareExpandDimsOutputStrategy() local
1363 int64_t axis_input = 1; in PrepareCumOutputStrategy() local
1418 int64_t axis_input = 1; in PrepareCumInputStrategy() local
1898 int64_t axis_input = GetValue<int64_t>(ops[i_ops]->input_value().at(1)); in PrepareExpandDimsInputStrategy() local