Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/src/
Dlite_kernel.h276 auto &ms_tensors = kernel_->inputs(); in in_tensors() local
291 auto &ms_tensors = kernel_->outputs(); in out_tensors() local
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_session.cc442 for (auto &ms_tensors : output_node_map_) { in RunGraph() local
495 …for (auto &ms_tensors : eval_output_node_map_) { // Allow to look at prediction also during train… in Train() local
527 for (auto &ms_tensors : eval_output_node_map_) { in Eval() local