Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dtensor_shape.cc68 TensorShape::TensorShape(const std::initializer_list<dsize_t> &list) in TensorShape() function in mindspore::dataset::TensorShape
73 TensorShape::TensorShape(const std::vector<dsize_t> &list) in TensorShape() function in mindspore::dataset::TensorShape
78 TensorShape::TensorShape(const TensorShape &shape) in TensorShape() function in mindspore::dataset::TensorShape
85 TensorShape::TensorShape(py::list l) in TensorShape() function in mindspore::dataset::TensorShape
100 TensorShape::TensorShape(cv::MatSize cv_size, uint32_t type) in TensorShape() function in mindspore::dataset::TensorShape
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/proto/
Dtensor_shape.proto19 message TensorShape { message
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dredistribution_operator_infer.h33 using TensorShape = Shape; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dbucket_batch_by_length_node.cc35 …const std::map<std::string, std::pair<TensorShape, std::shared_ptr<Tensor>>> &pad_info, bool pad_t… in BucketBatchByLengthNode()
Dbatch_node.cc37 std::map<std::string, std::pair<TensorShape, std::shared_ptr<Tensor>>> pad_map) in BatchNode()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/
Ddatasets.h43 class TensorShape; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Ddatasets.h45 class TensorShape; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/
Dpybind_conversion.cc256 Status toPadInfo(py::dict value, std::map<std::string, std::pair<TensorShape, std::shared_ptr<Tenso… in toPadInfo()