Searched refs:GetInputDeviceType (Results 1 – 12 of 12) sorted by relevance
92 if (kernel_info->GetInputDeviceType(input_index) == kNumberTypeInt64 && in ReducePrecision()115 …if (check_sm && major_sm < RECOMMEND_SM && kernel_info->GetInputDeviceType(input_index) == kNumber… in CheckSM()149 if (kernel_info->GetInputDeviceType(input_index) != in GpuKernelAttrCheck()
40 if (candidate_kernel_info->GetInputDeviceType(i) != dst_type || in AlternativeKernelInfoForInput()44 …} else if (candidate_kernel_info->GetInputDeviceType(i) != AnfAlgo::GetInputDeviceDataType(node, i… in AlternativeKernelInfoForInput()81 AnfAlgo::GetInputDeviceDataType(node, index) != kernel_info->GetInputDeviceType(index)) { in CheckInputs()
57 TypeId GetInputDeviceType(size_t input_index) const;135 …kernel_build_info_->inputs_device_type_.emplace_back(kernel_build_info->GetInputDeviceType(index)); in KernelBuildInfoBuilder()
38 TypeId KernelBuildInfo::GetInputDeviceType(size_t input_index) const { in GetInputDeviceType() function in mindspore::kernel::KernelBuildInfo108 …output_buffer << "<" << ToShortString(GetInputDeviceType(index)) << "x" << GetInputFormat(index) <… in ToString()
148 if (kernel_info.GetInputDeviceType(i) != dtype) { in CPUKernelSingleAttrCheck()149 … MS_LOG(DEBUG) << "input index:" << i << ", kernel info type:" << kernel_info.GetInputDeviceType(i) in CPUKernelSingleAttrCheck()
63 if (kernel_build_info.GetInputDeviceType(input_index) != input_origin_type) { in MatchInferOutputDataType()161 if (kernel_build_info.GetInputDeviceType(input_index) == prev_device_type) { in UpdateCurMatchCounts()262 auto device_dtype = kernel_build_info->GetInputDeviceType(input_index); in TagRaiseReduce()389 std::vector<TypeId> output_type = {selected_kernel_info->GetInputDeviceType(input_index)}; in SetWeightFormat()
92 if (kernel_info_list[index]->GetInputDeviceType(0) == fix_precision_type && in UpdateKernelInfo()
77 …dropout_kernel_info_builder->SetInputsDeviceType({old_kernel_info->GetInputDeviceType(0), kNumberT… in PreProcess()
52 …if (selected_kernel_info->GetInputDeviceType(i) != alternative_kernel_info->GetInputDeviceType(i))… in CheckKernelInfo()178 output_type = {selected_kernel_info.GetInputDeviceType(input_index)}; in SetTensorDeviceInfo()
59 if (kernel_info->GetInputDeviceType(i) != kNumberTypeFloat16 || in CheckKernelBuildInfo()
341 operator_info << TypeIdLabel(kernel_build_Info->GetInputDeviceType(i)); in KernelNotSupportException()
1014 auto dtype = build_info->GetInputDeviceType(input_idx); in GetInputDeviceDataType()