Searched refs:node_to_tensor (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 252 KernelMapTensor *node_to_tensor) { in CreateNodeOutputTensors() argument
255 MS_EXCEPTION_IF_NULL(node_to_tensor); in CreateNodeOutputTensors()
266 …t = CreateNodeOutputTensors(cnode->input(i), graph, input_tensors, tensor_to_node, node_to_tensor); in CreateNodeOutputTensors()
278 const auto &iter = node_to_tensor->find(item_with_index); in CreateNodeOutputTensors()
279 if (iter != node_to_tensor->end()) { in CreateNodeOutputTensors()
284 (*node_to_tensor)[item_with_index] = tensor; in CreateNodeOutputTensors()
1611 KernelMapTensor node_to_tensor; in UpdateOutputs() local
1616 …_back(CreateNodeOutputTensors(item, kernel_graph, input_tensors, tensor_to_node, &node_to_tensor)); in UpdateOutputs()
1697 KernelMapTensor node_to_tensor; in CreateOutputTensors() local
1701 …_back(CreateNodeOutputTensors(item, kernel_graph, input_tensors, tensor_to_node, &node_to_tensor)); in CreateOutputTensors()
[all …]
|
| D | kernel_graph.h | 283 …id SetOutputNodeToTensor(const KernelMapTensor &node_to_tensor) { output_node_to_tensor_ = node_to… in SetOutputNodeToTensor() argument
|