Searched defs:SelectKernel (Results 1 – 9 of 9) sorted by relevance
47 void SelectKernel(const CNodePtr &cnode) override { in SelectKernel() function in mindspore::opt::MockInsertTransOpKernelSelectTrans4Dto5D75 void SelectKernel(const CNodePtr &cnode) override { in SelectKernel() function in mindspore::opt::MockTransdataSplitKernelSelect
36 virtual void SelectKernel(const CNodePtr &cnode) { device::ascend::SelectKernelInfo(cnode); } in SelectKernel() function
54 void SelectKernel(const CNodePtr &cnode) override { in SelectKernel() function in mindspore::opt::MockInsertTransOpKernelSelectTrans4Dto5D
93 void SelectKernel(const CNodePtr &cnode) override { in SelectKernel() function in mindspore::opt::MockInsertTransOpKernelSelectTrans4Dto5D
100 void SelectKernel(const CNodePtr &cnode) override { in SelectKernel() function in mindspore::opt::MockRemoveInternalOutputTransOpKernelSelect
57 void SelectKernel(const CNodePtr &cnode) override { in SelectKernel() function in mindspore::opt::MockEliminate5To4And4To5KernelSelect
273 bool SelectKernel(const CNodePtr &kernel_node, KernelAttr *selected_kernel_attr, in SelectKernel() function
136 void GPUSession::SelectKernel(const std::shared_ptr<KernelGraph> &kernel_graph) const { in SelectKernel() function in mindspore::session::gpu::GPUSession
1182 void AscendSession::SelectKernel(const KernelGraph &kernel_graph) const { in SelectKernel() function in mindspore::session::AscendSession1789 void AscendSession::SelectKernel(NotNull<KernelGraphPtr> root_graph) { in SelectKernel() function in mindspore::session::AscendSession