Searched defs:tensor_to_node (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/graph/ |
| D | graph_impl.h | 136 std::map<tensor::TensorPtr, session::KernelWithIndex> tensor_to_node; in GetModelOutputsInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | cpu_kernel_runtime.cc | 245 std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in GetOrCreateTensorForOutput()
284 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateOutputTensors()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | session_basic.cc | 129 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateNodeOutputTensor()
381 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in CreateNodeOutputTensors()
1034 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in CreateOutputTensors()
1050 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors()
1128 std::map<tensor::TensorPtr, session::KernelWithIndex> tensor_to_node; in GetModelOutputsInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/ |
| D | cpu_session.cc | 162 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in CreateOutputTensors()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/ |
| D | gpu_session.cc | 545 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 329 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node) { in RunOpAssignMemory()
454 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in RunOpAssignOutputMemory()
|