Home
last modified time | relevance | path

Searched defs:SetKernelInfo (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dgraph_kernel_info.h34 virtual void SetKernelInfo(const CNodePtr &, KernelType) {} in SetKernelInfo() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/expander/fallback/
Dfallback_irbuilder.cc55 void SetKernelInfo(const AnfNodePtr &node) { in SetKernelInfo() function in mindspore::expander::InferHostAndDevice
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/common/
Ddevice_common_test.h168 void SetKernelInfo(const CNodePtr &node) const { in SetKernelInfo() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_session.cc246 void CPUSession::SetKernelInfo(const KernelGraph *kernel_graph) const { in SetKernelInfo() function in mindspore::session::CPUSession
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dkernel_build_utils.cc387 void SetKernelInfo(const CNodePtr &kernel_node) { in SetKernelInfo() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dkernel_info_setter.cc751 void GPUGraphKernelInfo::SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo() function in mindspore::device::gpu::GPUGraphKernelInfo
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dkernel_select_ascend.cc788 void SetKernelInfo(const CNodePtr &kernel_node, KernelType) override { in SetKernelInfo() function in mindspore::device::ascend::AscendGraphKernelInfo
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dkernel_select_cpu.cc757 void CPUGraphKernelInfo::SetKernelInfo(const CNodePtr &kernel_node, KernelType) { in SetKernelInfo() function in mindspore::device::cpu::CPUGraphKernelInfo
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_graph_executor.cc431 void SetKernelInfo(const AnfNodePtr &node) { in SetKernelInfo() function