Home
last modified time | relevance | path

Searched refs:tensor_shape_ (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_layout.cc36 buffer << std::endl << std::string("tensor shape = " + tensor_shape_.ToString()); in StandardToString()
55 tensor_shape_ = tensor_shape; in Init()
116 if (tensor_shape_.GetDimByIdx(i) % divisor != 0) { in TensorShapeDimensionIsDividedBySplitDeviceDimension()
146 tensor_shape_ = tensor_shape_origin_; in RemoveElementEqualToOneInDeviceArrangement()
186 …std::shared_ptr<std::vector<Arrangement>> expand_list_ptr = tensor_shape_.GetExpandShapeList(tenso… in ComputeArrangementByExpandedShape()
217 tensor_shape_.GetExpandShapeListPair(expanded_shape); in ExpandTensorShapeWithoutExtendDeviceArrangement()
287 tensor_shape_.GetExpandedShapeByExpandListReserveLeft(*re_map_shape_list_ptr); in ExpandDeviceArrangement()
301 Status status = ExpandShape(tensor_shape_.array(), expand_shape.array(), &in_expand_shape_shape); in TensorShapeCanBeExpanded()
305 return (in_expand_shape_shape == tensor_shape_.array()); in TensorShapeCanBeExpanded()
310 Status status = ExpandShape(tensor_shape_.array(), expand_shape.array(), &in_expand_shape_shape); in ComputeExpandedTensorShape()
[all …]
Dconstruct_operator.cc47 …int64_t prod_expect = std::accumulate(tensor_shape_.begin(), tensor_shape_.end(), 1, std::multipli… in ReshapeOP()
112 size_t size = tensor_shape_.size(); in StridedSliceOP()
117 for (auto num : tensor_shape_) { in StridedSliceOP()
165 if (LongToSize(concat_dim) >= tensor_shape_.size() || concat_dim < 0) { in ConcatOP()
209 if (tensor_shape_[LongToSize(split_dim)] % split_count != 0) { in AlltoAllOP()
214 if (LongToSize(concat_dim) >= tensor_shape_.size() || concat_dim < 0) { in AlltoAllOP()
Dtensor_layout.h65 Arrangement tensor_shape() const { return tensor_shape_; } in tensor_shape()
74 return (tensor_shape_ == tensor_layout.tensor_shape()); in IsSameTensorShape()
140 Arrangement tensor_shape_; variable
Dconstruct_operator.h46 void UpdateTensorShape(const Shape &tensor_shape) { tensor_shape_ = tensor_shape; } in UpdateTensorShape()
51 Shape tensor_shape_; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/
Ddata_schema.cc49 …: type_(DataType::DE_UNKNOWN), rank_(0), tensor_impl_(TensorImpl::kNone), tensor_shape_(nullptr) {} in ColDescriptor()
59 tensor_shape_ = std::make_unique<TensorShape>(*in_shape); in ColDescriptor()
61 tensor_shape_ = nullptr; in ColDescriptor()
76 tensor_shape_ = in_cd.HasShape() ? std::make_unique<TensorShape>(in_cd.Shape()) : nullptr; in ColDescriptor()
87 tensor_shape_ = in_cd.HasShape() ? std::make_unique<TensorShape>(in_cd.Shape()) : nullptr; in operator =()
99 if (tensor_shape_) { in Print()
100 out << *tensor_shape_ << ")\n"; in Print()
115 if (tensor_shape_ == nullptr) { in MaterializeTensorShape()
128 std::vector<dsize_t> requested_shape = tensor_shape_->AsVector(); in MaterializeTensorShape()
177 if (tensor_shape_ != nullptr) { in Shape()
[all …]
Ddata_schema.h100 bool HasShape() const { return tensor_shape_ != nullptr; } in HasShape()
110 std::unique_ptr<TensorShape> tensor_shape_; // The fixed shape (if given by user) variable