Home
last modified time | relevance | path

Searched refs:kernel_build_Info (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc332 auto kernel_build_Info = kernel_info->select_kernel_build_info(); in KernelNotSupportException() local
333 if (kernel_build_Info == nullptr) { in KernelNotSupportException()
337 size_t input_num = kernel_build_Info->GetInputNum(); in KernelNotSupportException()
341 operator_info << TypeIdLabel(kernel_build_Info->GetInputDeviceType(i)); in KernelNotSupportException()
348 size_t output_num = kernel_build_Info->GetOutputNum(); in KernelNotSupportException()
352 operator_info << TypeIdLabel(kernel_build_Info->GetOutputDeviceType(i)); in KernelNotSupportException()
353 if (i != kernel_build_Info->GetOutputNum() - 1) { in KernelNotSupportException()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel_factory.cc49 const KernelBuildInfo *kernel_build_Info = kernel_info->select_kernel_build_info(); in Create() local
50 MS_EXCEPTION_IF_NULL(kernel_build_Info); in Create()
51 std::pair<bool, size_t> ret_pair = CPUKernelAttrCheck(kernel_name, *kernel_build_Info); in Create()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/
Dgpu_kernel_factory.cc183 const KernelBuildInfo *kernel_build_Info = kernel_info->select_kernel_build_info(); in Create() local
184 MS_EXCEPTION_IF_NULL(kernel_build_Info); in Create()
185 std::pair<bool, size_t> ret_pair = GpuKernelAttrCheck(kernel_name, kernel_build_Info); in Create()