Home
last modified time | relevance | path

Searched defs:CreateTensor (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore-src/source/include/api/
Dtypes.h364 MSTensor *MSTensor::CreateTensor(const std::string &name, enum DataType type, const std::vector<int… in CreateTensor() function
369 MSTensor *MSTensor::CreateTensor(const std::string &name, const MSTensor &tensor, const std::string… in CreateTensor() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dseed_adapter.cc38 tensor::TensorPtr CreateTensor(T seed) { in CreateTensor() function
Dmax_pool3d_grad_grad_fission.cc34 tensor::TensorPtr CreateTensor(const AnfNodePtr &node) { in CreateTensor() function
Dspace_to_depth_split.cc32 tensor::TensorPtr CreateTensor(const AnfNodePtr &node) { in CreateTensor() function
/third_party/mindspore/mindspore-src/source/mindspore/core/utils/
Dtensor_construct_utils.cc85 tensor::TensorPtr TensorConstructUtils::CreateTensor(const TypePtr &type, const std::vector<int64_t… in CreateTensor() function in mindspore::TensorConstructUtils
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_fill.cc88 tensor::TensorPtr CreateTensor(float value, TypePtr dtype) { in CreateTensor() function
Dtest_ops_concat.cc123 tensor::TensorPtr CreateTensor(const ShapeVector &shape, std::vector<float> value) { in CreateTensor() function
Dtest_ops_reduce_extand.cc61 tensor::TensorPtr CreateTensor(const ShapeVector &shape, const TypeId &dtype, std::vector<T> value)… in CreateTensor() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/common/
Dcommon_test.cc33 lite::Tensor *CommonTest::CreateTensor(TypeId dtype, std::vector<int> shape, const Format &format, in CreateTensor() function in mindspore::CommonTest
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/
Dtypes.cc113 MSTensor *MSTensor::CreateTensor(const std::vector<char> &name, enum DataType type, const std::vect… in CreateTensor() function in mindspore::MSTensor
203 MSTensor *MSTensor::CreateTensor(const std::vector<char> &name, const MSTensor &tensor, const std::… in CreateTensor() function in mindspore::MSTensor
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dreduce_axis_update.cc37 tensor::TensorPtr CreateTensor(const std::vector<int64_t> &values) { in CreateTensor() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/
Dtypes.cc143 MSTensor *MSTensor::CreateTensor(const std::vector<char> &name, enum DataType type, const std::vect… in CreateTensor() function in mindspore::MSTensor
160 MSTensor *MSTensor::CreateTensor(const std::vector<char> &name, const MSTensor &tensor, const std::… in CreateTensor() function in mindspore::MSTensor
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/
Dpyboost_utils.cc37 void CreateTensor(const TypePtr &type, const ShapeVector &shape_vector, const AbstractBasePtr &abst… in CreateTensor() function
46 void CreateTensor(const TypePtr &type, const ShapeVector &shape_vector, std::vector<tensor::BaseTen… in CreateTensor() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/
Dtensor.cc536 Tensor *Tensor::CreateTensor(const std::string &name, TypeId type, const std::vector<int> &shape, c… in CreateTensor() function in mindspore::lite::Tensor
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_export.cc154 std::unique_ptr<schema::TensorT> TrainExport::CreateTensor( in CreateTensor() function in mindspore::lite::TrainExport
/third_party/mindspore/mindspore-src/source/mindspore/lite/schema/inner/
Dmodel_generated.h3113 inline ::flatbuffers::Offset<Tensor> CreateTensor(::flatbuffers::FlatBufferBuilder &_fbb, const Ten… in CreateTensor() function