Searched refs:SetKernelInfo (Results 1 – 19 of 19) sorted by relevance
121 SetKernelInfo(graph.get()); in CompileGraphImpl()235 SetKernelInfo(kernel_graph.get()); in BuildOpImpl()313 void CPUSession::SetKernelInfo(const KernelGraph *kernel_graph) { in SetKernelInfo() function in mindspore::session::CPUSession318 device::cpu::SetKernelInfo(kernel_node); in SetKernelInfo()
60 void SetKernelInfo(const KernelGraph *kernel_graph);
141 device::gpu::SetKernelInfo(kernel_node); in SelectKernel()
34 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type);
544 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo() function
51 device::gpu::SetKernelInfo(node_list[i]); in KernelSelect()
70 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type = KernelType::UNKNOWN_KERNEL…
312 SetKernelInfo(cnode, KernelType::AKG_KERNEL); in SetGraphKernelInfo()398 void SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo() function
214 SetKernelInfo(node); in CreateStreamSwitchNode()
31 void SetKernelInfo(const CNodePtr &apply_kernel_ptr);
302 void SetKernelInfo(const CNodePtr &kernel_node) { in SetKernelInfo() function
93 device::gpu::SetKernelInfo(fused_batch_norm_with_relu); in Process()
98 device::gpu::SetKernelInfo(fused_batch_norm_with_add_relu); in Process()
106 device::gpu::SetKernelInfo(fused_batch_norm_grad_with_relu); in Process()
200 device::gpu::SetKernelInfo(fused_batch_norm_add_relu_grad); in Process()
176 SetKernelInfo(node); in SetOperatorInfo()
359 SetKernelInfo(node); in SetOperatorInfo()
608 device::ascend::SetKernelInfo(cnode, kernel_type); in ResetKernelInfo()611 device::gpu::SetKernelInfo(cnode, kernel_type); in ResetKernelInfo()