Searched refs:GetOutputNum (Results 1 – 10 of 10) sorted by relevance
47 if (kernel_info->GetOutputNum() != iter_second->at(attr_index).first.GetOutputSize()) { in CheckIOParam()100 for (size_t output_index = 0; output_index < kernel_info->GetOutputNum(); output_index++) { in ReducePrecision()159 if (kernel_info->GetOutputNum() > 0) { in GpuKernelAttrCheck()163 for (size_t output_index = 0; output_index < kernel_info->GetOutputNum(); output_index++) { in GpuKernelAttrCheck()
44 return kernel_build_info->GetOutputNum() == output_tensor_num && in FilterInvalidKernelInfo()55 if (kernel_info->GetOutputNum() != output_tensor_num) { in FilterInvalidKernelInfo()57 << " cannot match the kernel's output size [" << kernel_info->GetOutputNum() << "]"; in FilterInvalidKernelInfo()
95 size_t GetOutputNum() const;140 for (size_t index = 0; index < kernel_build_info->GetOutputNum(); ++index) { in KernelBuildInfoBuilder()
66 size_t KernelBuildInfo::GetOutputNum() const { return outputs_format_.size(); } in GetOutputNum() function in mindspore::kernel::KernelBuildInfo111 for (size_t index = 0; index < GetOutputNum(); ++index) { in ToString()
51 for (size_t i = 0; i < candidate_kernel_info->GetOutputNum(); i++) { in AlternativeKernelInfoForInput()91 if (AnfAlgo::GetOutputTensorNum(node) != kernel_info->GetOutputNum()) { in CheckOtherOutputs()94 for (size_t index = 0; index < kernel_info->GetOutputNum(); ++index) { in CheckOtherOutputs()
348 size_t output_num = kernel_build_Info->GetOutputNum(); in KernelNotSupportException()353 if (i != kernel_build_Info->GetOutputNum() - 1) { in KernelNotSupportException()
64 for (size_t i = 0; i < kernel_info->GetOutputNum(); ++i) { in CheckKernelBuildInfo()
57 size_t selected_output_num = selected_kernel_info->GetOutputNum(); in CheckKernelInfo()58 size_t alternative_output_num = alternative_kernel_info->GetOutputNum(); in CheckKernelInfo()
154 for (size_t i = 0; i < kernel_info.GetOutputNum(); ++i) { in CPUKernelSingleAttrCheck()
68 for (size_t output_index = 0; output_index < kernel_build_info.GetOutputNum(); ++output_index) { in MatchInferOutputDataType()271 for (size_t output_index = 0; output_index < kernel_build_info->GetOutputNum(); ++output_index) { in TagRaiseReduce()