Home
last modified time | relevance | path

Searched defs:ms_tensors (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dtensor_utils.cc81 …ore::tensor::TensorPtr> TensorUtils::MSTensorToTensorPtr(const std::vector<MSTensor> &ms_tensors) { in MSTensorToTensorPtr()
101 std::vector<MSTensor> ms_tensors; in TensorPtrToMSTensor() local
112 …mindspore::tensor::Tensor> TensorUtils::MSTensorToTensor(const std::vector<MSTensor> &ms_tensors) { in MSTensorToTensor()
137 std::vector<MSTensor> ms_tensors; in TensorToMSTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/
Dkernel_exec.h272 auto &ms_tensors = kernel_->inputs(); in in_tensors() local
292 auto &ms_tensors = kernel_->outputs(); in out_tensors() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/model/acl/acl_vm/
Dacl_vm.cc40 std::vector<MSTensor> ms_tensors; in ParseVectorMsTensorRef() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/
Dtrain_session.cc502 for (auto &ms_tensors : output_node_map_) { in RunGraph() local
555 …for (auto &ms_tensors : eval_output_node_map_) { // Allow to look at prediction also during train… in Train() local
587 for (auto &ms_tensors : eval_output_node_map_) { in Eval() local