Home
last modified time | relevance | path

Searched defs:SetShape (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Dde_tensor.h47 void SetShape(const std::vector<int64_t> &shape) override { shape_ = shape; }; in SetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/common/
Dcommon_test.h45 void SetShape(int *dst, std::vector<int> src, size_t num) { in SetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/
Dtypes.cc85 void SetShape(const std::vector<int64_t> &shape) override { shape_ = shape; } in SetShape() function in mindspore::TensorDefaultImpl
119 void SetShape(const std::vector<int64_t> &shape) override { shape_ = shape; } in SetShape() function in mindspore::TensorReferenceImpl
460 void MSTensor::SetShape(const std::vector<int64_t> &) { MS_LOG_EXCEPTION << "Invalid implement."; } in SetShape() function in mindspore::MSTensor
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensor_info.cc124 void TensorInfo::SetShape(const std::vector<int64_t> &shape) { in SetShape() function in mindspore::lite::TensorInfo
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dmatmul_infer.c65 int SetShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, size_t outputs_si… in SetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/tensor/
Dtensor_impl.h159 void SetShape(const std::vector<int64_t> &shape) override { in SetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/python/src/
Dtensor_numpy_impl.h56 void SetShape(const std::vector<int64_t> &shape) override { in SetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/
Dpass_utils.cc47 bool SetShape(const Tensor *src_tensor, Tensor *dst_tensor) { in SetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dtensor_default_impl.h84 void SetShape(const std::vector<int64_t> &shape) override { shape_ = shape; } in SetShape() function
Dtensor_utils.h151 …void SetShape(const std::vector<int64_t> &) override { MS_LOG_EXCEPTION << "Cannot set shape for T… in SetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dacl_utils.h126 AclTensorDescMaker &SetShape(const ShapeVector &shape) { in SetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/debug/
Dtensor_data.h215 void SetShape(const std::vector<int64_t> &shape) { this->shape_ = shape; } in SetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dkernel.cc192 void KernelTensor::SetShape(const abstract::BaseShapePtr &shape) { in SetShape() function in mindspore::kernel::KernelTensor
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/
Dtypes.cc590 void MSTensor::SetShape(const std::vector<int64_t> &shape) { in SetShape() function in mindspore::MSTensor