Searched refs:selected_kernel_info (Results 1 – 2 of 2) sorted by relevance
40 const std::shared_ptr<KernelBuildInfo> &selected_kernel_info) { in CheckKernelInfo() argument41 MS_EXCEPTION_IF_NULL(selected_kernel_info); in CheckKernelInfo()43 size_t selected_input_num = selected_kernel_info->GetInputNum(); in CheckKernelInfo()49 if (selected_kernel_info->GetInputFormat(i) != alternative_kernel_info->GetInputFormat(i)) { in CheckKernelInfo()52 …if (selected_kernel_info->GetInputDeviceType(i) != alternative_kernel_info->GetInputDeviceType(i))… in CheckKernelInfo()57 size_t selected_output_num = selected_kernel_info->GetOutputNum(); in CheckKernelInfo()63 if (selected_kernel_info->GetOutputFormat(i) != alternative_kernel_info->GetOutputFormat(i)) { in CheckKernelInfo()66 …if (selected_kernel_info->GetOutputDeviceType(i) != alternative_kernel_info->GetOutputDeviceType(i… in CheckKernelInfo()103 …Kernel(const CNodePtr &kernel_node, const std::shared_ptr<KernelBuildInfo> &selected_kernel_info) { in SelectAkgKernel() argument105 MS_EXCEPTION_IF_NULL(selected_kernel_info); in SelectAkgKernel()[all …]
384 auto selected_kernel_info = AnfAlgo::GetSelectKernelBuildInfo(kernel_node); in SetWeightFormat() local385 MS_EXCEPTION_IF_NULL(selected_kernel_info); in SetWeightFormat()389 std::vector<TypeId> output_type = {selected_kernel_info->GetInputDeviceType(input_index)}; in SetWeightFormat()434 auto selected_kernel_info = AnfAlgo::GetSelectKernelBuildInfo(kernel_node); in SetTensorDeviceInfo() local435 MS_EXCEPTION_IF_NULL(selected_kernel_info); in SetTensorDeviceInfo()444 …if (RefreshCastAndParamWeightFormat(real_input_node, selected_kernel_info->GetInputFormat(input_in… in SetTensorDeviceInfo()450 auto refresh_format = selected_kernel_info->GetInputFormat(input_index); in SetTensorDeviceInfo()464 std::shared_ptr<kernel::KernelBuildInfo> selected_kernel_info = nullptr; in SetMatchedKernelInfo() local469 selected_kernel_info = ChooseMatchedKernelInfo(kernel_node, filtered_kernel_info_list); in SetMatchedKernelInfo()475 selected_kernel_info = ChooseMatchedKernelInfo(kernel_node, filtered_kernel_info_list); in SetMatchedKernelInfo()[all …]