Home
last modified time | relevance | path

Searched defs:single_shape (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dalltoall_fusion.cc121 auto single_shape = AnfAlgo::GetOutputDetailShape(split_outputs[0], 0); in CreateAllToAllvNode() local
170 auto single_shape = common::AnfAlgo::GetOutputInferShape(all_to_all_v_outputs[0], 0); in CreateConcatNode() local
Dneighbor_exchange_v2_fusion.cc598 auto single_shape = in CreateLeftRightConcat() local
632 auto single_shape = common::AnfAlgo::GetOutputInferShape(neighbor_exchange_v2_input, 0); in CreateMiddleConcat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dall_to_all_unify_mindir.cc119 auto single_shape = AnfAlgo::GetOutputDetailShape(split_outputs[0], 0UL); in CreateAllToAllvNode() local
191 auto single_shape = AnfAlgo::GetOutputDetailShape(split_outputs[0], 0UL); in CreateAllToAllvNode() local
263 auto single_shape = common::AnfAlgo::GetOutputInferShape(input_node_outputs[0], 0); in CreateConcatNode() local
Dneighbor_exchange_v2_unify_mindir.cc525 auto single_shape = common::AnfAlgo::GetOutputInferShape(all_to_all_v_outputs[output_idx], 0UL); in CreateLeftRightConcat() local
556 auto single_shape = common::AnfAlgo::GetOutputInferShape(neighbor_exchange_v2_input, 0UL); in CreateMiddleConcat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc2247 const auto &single_shape = std::make_shared<abstract::Shape>(single_shape_vector); in SequenceToTensor() local