Searched refs:SetKernelMod (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/ |
| D | akg_ascend_kernel_build.cc | 44 AnfAlgo::SetKernelMod(kernel_mod_ptr, anf_node.get()); in AkgSetKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/ |
| D | akg_gpu_kernel_build.cc | 46 AnfAlgo::SetKernelMod(kernel_mod_ptr, anf_node.get()); in AkgSetKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_build.cc | 68 … session::AnfRuntimeAlgorithm::SetKernelMod((kernel::KernelModPtr)gpu_kernel_ptr, kernel.get()); in CreateGPUKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 199 AnfAlgo::SetKernelMod(kernel_mod, cur_node.get()); in TaskFinishProcess()
279 AnfAlgo::SetKernelMod(kernel_mod_ptr, node); in SearchInCache()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/ |
| D | insert_cast_cpu.cc | 65 AnfAlgo::SetKernelMod(cpu_kernel, cast.get()); in AddCastOpNodeToGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_device_context.cc | 204 AnfAlgo::SetKernelMod(cpu_kernel, node.get()); in CreateKernel()
|
| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 711 TEST_F(AnfRuntimeAlgorithmTest, SetKernelMod) { in TEST_F() argument
716 AnfAlgo::SetKernelMod(nullptr, add.get()); in TEST_F()
717 EXPECT_THROW(AnfAlgo::SetKernelMod(nullptr, nullptr), std::runtime_error); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | cpu_session.cc | 381 AnfAlgo::SetKernelMod(cpu_kernel, kernel_node.get()); in BuildKernel()
|
| D | anf_runtime_algorithm.h | 234 static void SetKernelMod(const kernel::KernelModPtr &kernel_mod, AnfNode *node);
|
| D | anf_runtime_algorithm.cc | 1395 void AnfRuntimeAlgorithm::SetKernelMod(const KernelModPtr &kernel_mod, AnfNode *node) { in SetKernelMod() function in mindspore::session::AnfRuntimeAlgorithm
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 126 AnfAlgo::SetKernelMod(kernel_mod_ptr, anf_node.get()); in KernelBuildParallelCompile()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | ub_pattern_fusion.cc | 525 AnfAlgo::SetKernelMod(kernel_ptr, buffer_fusion.get()); in ReplaceFusionOp()
|