Home
last modified time | relevance | path

Searched defs:expand_shape (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dcdist_fission.cc66 auto expand_shape = AnfAlgo::GetOutputInferShape(input_node, 0); in AddBroadCastToNode() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Darrangement.cc92 Shape expand_shape = expand_list[i].GetFrontElementByValue(GetDimByIdx(i)); in GetExpandedShapeByExpandListRemoveLeft() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Ddecrease_transpose_algo.cc174 ShapeVector expand_shape(data_info.shape_.begin(), data_info.shape_.end()); in ConvertTensorToNCOrNH() local