Searched refs:kernel_mod_ptr (Results 1 – 6 of 6) sorted by relevance
39 auto kernel_mod_ptr = std::make_shared<AkgKernelMod>(kernel_pack); in AkgSetKernelMod() local41 kernel_mod_ptr->SetInputSizeList(json_generator.input_size_list()); in AkgSetKernelMod()42 kernel_mod_ptr->SetOutputSizeList(json_generator.output_size_list()); in AkgSetKernelMod()43 kernel_mod_ptr->SetWorkspaceSizeList(kernel_json_info.workspaces); in AkgSetKernelMod()44 AnfAlgo::SetKernelMod(kernel_mod_ptr, anf_node.get()); in AkgSetKernelMod()
41 auto kernel_mod_ptr = std::make_shared<GpuKernelMod>(kernel_pack); in AkgSetKernelMod() local43 kernel_mod_ptr->SetInputSizeList(json_generator.input_size_list()); in AkgSetKernelMod()44 kernel_mod_ptr->SetOutputSizeList(json_generator.output_size_list()); in AkgSetKernelMod()45 kernel_mod_ptr->SetWorkspaceSizeList(kernel_json_info.workspaces); in AkgSetKernelMod()46 AnfAlgo::SetKernelMod(kernel_mod_ptr, anf_node.get()); in AkgSetKernelMod()
77 …onst std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<AicpuOpKernelMod> &kernel_mod_ptr) { in SetIOSize() argument79 MS_EXCEPTION_IF_NULL(kernel_mod_ptr); in SetIOSize()88 kernel_mod_ptr->SetInputSizeList(input_size_list); in SetIOSize()101 kernel_mod_ptr->SetOutputSizeList(output_size_list); in SetIOSize()267 const std::shared_ptr<AicpuOpKernelMod> &kernel_mod_ptr) { in CreateNodeDefBytes() argument268 MS_EXCEPTION_IF_NULL(kernel_mod_ptr); in CreateNodeDefBytes()278 kernel_mod_ptr->SetNodeDef(nodeDefStr); in CreateNodeDefBytes()362 …onst std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<AicpuOpKernelMod> &kernel_mod_ptr) { in CreateExtInfo() argument364 MS_EXCEPTION_IF_NULL(kernel_mod_ptr); in CreateExtInfo()403 kernel_mod_ptr->SetExtInfo(ext_info); in CreateExtInfo()[all …]
277 auto kernel_mod_ptr = GenKernelMod(input_size_list, output_size_list, cached_kernel_pack); in SearchInCache() local278 MS_EXCEPTION_IF_NULL(kernel_mod_ptr); in SearchInCache()279 AnfAlgo::SetKernelMod(kernel_mod_ptr, node); in SearchInCache()291 auto kernel_mod_ptr = std::make_shared<TbeKernelMod>(kernel_pack); in GenKernelMod() local292 MS_EXCEPTION_IF_NULL(kernel_mod_ptr); in GenKernelMod()293 kernel_mod_ptr->SetInputSizeList(input_size_list); in GenKernelMod()294 kernel_mod_ptr->SetOutputSizeList(output_size_list); in GenKernelMod()295 kernel_mod_ptr->SetWorkspaceSizeList(kernel_json_info.workspaces); in GenKernelMod()296 return kernel_mod_ptr; in GenKernelMod()
47 kernel::KernelModPtr kernel_mod_ptr = nullptr; in SerialCompileImpl() local51 kernel_mod_ptr = kernel::AicpuOpBuild(anf_node); in SerialCompileImpl()55 kernel_mod_ptr = kernel::HostOpBuild(anf_node); in SerialCompileImpl()59 kernel_mod_ptr = kernel::RtOpBuild(anf_node); in SerialCompileImpl()63 kernel_mod_ptr = kernel::HcclOpBuild(anf_node); in SerialCompileImpl()71 return kernel_mod_ptr; in SerialCompileImpl()124 kernel::KernelModPtr kernel_mod_ptr = SerialCompileImpl(anf_node); in KernelBuildParallelCompile() local125 MS_EXCEPTION_IF_NULL(kernel_mod_ptr); in KernelBuildParallelCompile()126 AnfAlgo::SetKernelMod(kernel_mod_ptr, anf_node.get()); in KernelBuildParallelCompile()
92 const std::shared_ptr<kernel::AicpuOpKernelMod> &kernel_mod_ptr) { in SetNodedefProto() argument94 MS_ERROR_IF_NULL(kernel_mod_ptr); in SetNodedefProto()105 kernel_mod_ptr->SetNodeDef(nodeDefStr); in SetNodedefProto()