Home
last modified time | relevance | path

Searched defs:TensorShape (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Dtensor_shape.cc64 TensorShape::TensorShape(const std::initializer_list<dsize_t> &list) { AddListToShape(list); } in TensorShape() function in mindspore::dataset::TensorShape
66 TensorShape::TensorShape(const std::vector<dsize_t> &list) { AddListToShape(list); } in TensorShape() function in mindspore::dataset::TensorShape
68 TensorShape::TensorShape(const TensorShape &shape) in TensorShape() function in mindspore::dataset::TensorShape
71 TensorShape::TensorShape(TensorShape &&shape) noexcept in TensorShape() function in mindspore::dataset::TensorShape
93 TensorShape::TensorShape(py::list l) { in TensorShape() function in mindspore::dataset::TensorShape
107 TensorShape::TensorShape(cv::MatSize cv_size, uint32_t type) : known_(true) { in TensorShape() function in mindspore::dataset::TensorShape
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Ddshape.h148 TensorShape() : shape_() {} in TensorShape() function
153 TensorShape(const std::initializer_list<ShapeValueDType> &list) : shape_(list) {} in TensorShape() function
158 explicit TensorShape(const ShapeVector &list) : shape_(list) {} in TensorShape() function
163 explicit TensorShape(ShapeVector &&list) : shape_(std::move(list)) {} in TensorShape() function
169TensorShape(const ShapeVector &list, const ShapeVector &max_shape) : shape_(list), max_shape_(max_… in TensorShape() function
173 MS_DECLARE_PARENT(TensorShape, BaseShape) in MS_DECLARE_PARENT() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dnon_zero_impl.cu22 struct TensorShape { struct
23 size_t data[kNonZeroMaxDim];
24 size_t rank;
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/proto/
Dtensor_shape.proto19 message TensorShape { message
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dredistribution_operator_infer.h35 using TensorShape = Shape; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dbucket_batch_by_length_node.cc28 …const std::map<std::string, std::pair<TensorShape, std::shared_ptr<Tensor>>> &pad_info, bool pad_t… in BucketBatchByLengthNode()
Dbatch_node.cc36 std::map<std::string, std::pair<TensorShape, std::shared_ptr<Tensor>>> pad_map, in BatchNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/liteapi/include/
Ddatasets.h42 class TensorShape; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/aicpu_op_proto/
Daicpu_tensor.proto57 message TensorShape { message
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/include/dataset/
Ddatasets.h53 class TensorShape; variable
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/
Dbuiltin_operations.py188 def TensorShape(x): function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/python/
Dpybind_conversion.cc279 std::map<std::string, std::pair<TensorShape, std::shared_ptr<Tensor>>> *pad_info) { in toPadInfo()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.h2649 TensorShape() : BaseOperator(kNameTensorShape) {} in TensorShape() function