Home
last modified time | relevance | path

Searched refs:LiteTensorsToMSTensors (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/lite/src/cxx_api/model/
Dmodel_impl.cc158 std::vector<MSTensor> inputs = LiteTensorsToMSTensors(before_inputs); in RunGraph()
159 std::vector<MSTensor> outputs = LiteTensorsToMSTensors(before_outputs); in RunGraph()
169 std::vector<MSTensor> inputs = LiteTensorsToMSTensors(before_inputs); in RunGraph()
170 std::vector<MSTensor> outputs = LiteTensorsToMSTensors(before_outputs); in RunGraph()
358 std::vector<MSTensor> res = LiteTensorsToMSTensors(params, false); in GetGradients()
396 std::vector<MSTensor> res = LiteTensorsToMSTensors(params); in GetOptimizerParams()
/third_party/mindspore/mindspore/lite/src/cxx_api/
Dtensor_utils.h34 std::vector<MSTensor> MS_API LiteTensorsToMSTensors(const std::vector<mindspore::tensor::MSTensor *…
Dtensor_utils.cc65 std::vector<MSTensor> LiteTensorsToMSTensors(const std::vector<mindspore::tensor::MSTensor *> &srcT… in LiteTensorsToMSTensors() function
/third_party/mindspore/mindspore/lite/src/
Dkernel_registry.cc155 …auto base_kernel = creator(LiteTensorsToMSTensors(in_tensors), LiteTensorsToMSTensors(out_tensors), in GetCustomKernel()
Dscheduler.cc393 ms_inputs_ = LiteTensorsToMSTensors(inputs_); in ReplaceDelegateKernels()
394 ms_outputs_ = LiteTensorsToMSTensors(outputs_); in ReplaceDelegateKernels()
/third_party/mindspore/mindspore/lite/src/common/
Dtensor_util.h50 std::vector<mindspore::MSTensor> LiteTensorsToMSTensors(const std::vector<lite::Tensor *> &lite_ten…
Dtensor_util.cc266 std::vector<mindspore::MSTensor> LiteTensorsToMSTensors(const std::vector<lite::Tensor *> &lite_ten… in LiteTensorsToMSTensors() function