Searched defs:tensor_to_node (Results 1 – 7 of 7) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/session/ |
D | pynative_task_manager.h | 35 std::map<tensor::TensorPtr, KernelWithIndex> tensor_to_node) in RunOpContext() 48 …const std::map<tensor::TensorPtr, KernelWithIndex> &tensor_to_node() const { return tensor_to_node… in tensor_to_node() function
|
D | cpu_session.cc | 160 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateOutputTensors() 259 …ion::UpdateDynamicOutputShape(const std::map<tensor::TensorPtr, KernelWithIndex> &tensor_to_node) { in UpdateDynamicOutputShape() 296 std::map<tensor::TensorPtr, session::KernelWithIndex> tensor_to_node; in RunOpImpl() local
|
D | ascend_session.cc | 834 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in LaunchFunc() 894 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in PrepareForOutputTensor() 957 std::map<tensor::TensorPtr, session::KernelWithIndex> tensor_to_node; in RunOpImpl() local 1014 std::map<tensor::TensorPtr, session::KernelWithIndex> tensor_to_node; in RunOpImplOrigin() local 1488 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in RunOpMemoryAllocNew() 2010 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors()
|
D | gpu_session.cc | 564 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors() 692 std::map<tensor::TensorPtr, session::KernelWithIndex> tensor_to_node; in RunOpImpl() local
|
D | session_basic.cc | 198 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateNodeOutputTensor() 251 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in CreateNodeOutputTensors() 1691 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateOutputTensors() 1712 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors() 1796 std::map<tensor::TensorPtr, session::KernelWithIndex> tensor_to_node; in GetModelOutputsInfo() local
|
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
D | cpu_kernel_runtime.cc | 188 std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreatTensorForOutput() 249 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreatTensorForOutput() 281 … std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateOutputTensors()
|
/third_party/mindspore/mindspore/ccsrc/runtime/device/ |
D | kernel_runtime.cc | 317 … const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node) { in RunOpAssignMemory() 437 …t AnfNodePtr &kernel, const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node)… in RunOpAssignOutputMemory()
|