Home
last modified time | relevance | path

Searched defs:tensor_to_node (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dpynative_task_manager.h35 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
Dcpu_session.cc160 … 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
Dascend_session.cc834 … 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()
Dgpu_session.cc564 … 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
Dsession_basic.cc198 … 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/
Dcpu_kernel_runtime.cc188 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/
Dkernel_runtime.cc317 … 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()