Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/
Dgather_fusion_mapper.cc37 auto axis_input = cnode->input(kNameGatherInputNum - 1); in Mapper() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_generate_strategy.cc211 auto axis_input = GetValue<int64_t>(ops[iter_ops]->input_value().at(2)); in PrepareGatherV2() local
254 auto axis_input = GetValue<int64_t>(ops[iter_ops]->input_value().at(2)); in PrepareGatherV2P() local