Home
last modified time | relevance | path

Searched refs:expand_shape (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Darrangement.cc92 Shape expand_shape = expand_list[i].GetFrontElementByValue(GetDimByIdx(i)); in GetExpandedShapeByExpandListRemoveLeft() local
93 if (expand_shape.empty()) { in GetExpandedShapeByExpandListRemoveLeft()
96 (void)new_shape.insert(new_shape.end(), expand_shape.begin(), expand_shape.end()); in GetExpandedShapeByExpandListRemoveLeft()
110 …<std::vector<Arrangement>> Arrangement::GetExpandShapeList(const Arrangement &expand_shape) const { in GetExpandShapeList()
115 for (size_t i = 0; i < expand_shape.GetDimSize(); i++) { in GetExpandShapeList()
116 size *= expand_shape.GetDimByIdx(i); in GetExpandShapeList()
121 shape.push_back(expand_shape.GetDimByIdx(i)); in GetExpandShapeList()
124 shape.push_back(expand_shape.GetDimByIdx(i)); in GetExpandShapeList()
142 const Arrangement &expand_shape) const { in GetExpandShapeListPair()
143 …std::shared_ptr<std::vector<Arrangement>> expand_shape_list_ptr = GetExpandShapeList(expand_shape); in GetExpandShapeListPair()
Darrangement.h38 …std::shared_ptr<std::vector<Arrangement>> GetExpandShapeList(const Arrangement &expand_shape) cons…
45 const Arrangement &expand_shape) const;
Dreshape_layout_transfer.cc122 const Arrangement &expand_shape) const { in ExpandFromTensorShapeAndExpandToDeviceArrangement()
123 …::shared_ptr<TensorLayout> extend_tensor_shape_from_ptr = from_in_.ExpandTensorShape(expand_shape); in ExpandFromTensorShapeAndExpandToDeviceArrangement()
Dtensor_layout.cc299 bool TensorLayout::TensorShapeCanBeExpanded(const Arrangement &expand_shape) const { in TensorShapeCanBeExpanded()
301 Status status = ExpandShape(tensor_shape_.array(), expand_shape.array(), &in_expand_shape_shape); in TensorShapeCanBeExpanded()
308 …_ptr<Arrangement> TensorLayout::ComputeExpandedTensorShape(const Arrangement &expand_shape) const { in ComputeExpandedTensorShape()
310 Status status = ExpandShape(tensor_shape_.array(), expand_shape.array(), &in_expand_shape_shape); in ComputeExpandedTensorShape()
Dreshape_layout_transfer.h36 const Arrangement &expand_shape) const;
Dtensor_layout.h89 std::shared_ptr<Arrangement> ComputeExpandedTensorShape(const Arrangement &expand_shape) const;
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dcdist_fission.cc66 auto expand_shape = AnfAlgo::GetOutputInferShape(input_node, 0); in AddBroadCastToNode() local
67 (void)expand_shape.insert(expand_shape.end() + dim, 1); in AddBroadCastToNode()
68 AnfAlgo::SetOutputInferTypeAndShape({dtype}, {expand_shape}, expand_dims.get()); in AddBroadCastToNode()
/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
176 expand_shape = {1, 1, 1, data_info.shape_[0]}; in ConvertTensorToNCOrNH()
178 expand_shape = {1, 1, data_info.shape_[0], data_info.shape_[1]}; in ConvertTensorToNCOrNH()
180 expand_shape = {1, data_info.shape_[0], data_info.shape_[1], data_info.shape_[kInputIndexTwo]}; in ConvertTensorToNCOrNH()
182 … tensor = std::make_shared<tensor::Tensor>(static_cast<TypeId>(data_info.data_type_), expand_shape, in ConvertTensorToNCOrNH()