Searched refs:info_builder (Results 1 – 9 of 9) sorted by relevance
49 kernel::KernelBuildInfo::KernelBuildInfoBuilder info_builder; in CreateReduceSum() local50 info_builder.SetInputsFormat({AnfAlgo::GetInputFormat(node, 0)}); in CreateReduceSum()51 info_builder.SetInputsDeviceType({kFloat32->type_id()}); in CreateReduceSum()52 info_builder.SetOutputsFormat({AnfAlgo::GetOutputFormat(node, 0)}); in CreateReduceSum()53 info_builder.SetOutputsDeviceType({kFloat32->type_id()}); in CreateReduceSum()54 info_builder.SetProcessor(AnfAlgo::GetProcessor(node)); in CreateReduceSum()55 info_builder.SetKernelType(KernelType::AKG_KERNEL); in CreateReduceSum()56 info_builder.SetFusionType(kernel::FusionType::OPAQUE); in CreateReduceSum()57 AnfAlgo::SetSelectKernelBuildInfo(info_builder.Build(), cnode.get()); in CreateReduceSum()
121 kernel::KernelBuildInfo::KernelBuildInfoBuilder info_builder; in SetNodeInfo() local122 info_builder.SetInputsFormat(node_io_info.inputs_format); in SetNodeInfo()123 info_builder.SetInputsDeviceType(node_io_info.inputs_type); in SetNodeInfo()124 info_builder.SetOutputsFormat(node_io_info.outputs_format); in SetNodeInfo()125 info_builder.SetOutputsDeviceType(node_io_info.outputs_type); in SetNodeInfo()126 info_builder.SetKernelType(AnfAlgo::GetKernelType(orig_node)); in SetNodeInfo()127 info_builder.SetOpPattern(AnfAlgo::GetOpPattern(orig_node)); in SetNodeInfo()128 info_builder.SetFusionType(AnfAlgo::GetFusionType(orig_node)); in SetNodeInfo()129 info_builder.SetProcessor(AnfAlgo::GetProcessor(orig_node)); in SetNodeInfo()130 AnfAlgo::SetSelectKernelBuildInfo(info_builder.Build(), new_node.get()); in SetNodeInfo()
737 kernel::KernelBuildInfo::KernelBuildInfoBuilder info_builder; in CreateCNode() local738 info_builder.SetInputsFormat(input_formats); in CreateCNode()739 info_builder.SetInputsDeviceType(input_types); in CreateCNode()740 info_builder.SetOutputsFormat(output_formats); in CreateCNode()741 info_builder.SetOutputsDeviceType(output_types); in CreateCNode()742 info_builder.SetProcessor(kernel::GetProcessorFromContext()); in CreateCNode()743 info_builder.SetKernelType(KernelType::AKG_KERNEL); in CreateCNode()744 info_builder.SetFusionType(kernel::FusionType::OPAQUE); in CreateCNode()745 auto selected_info = info_builder.Build(); in CreateCNode()
61 auto info_builder = in SetCastFormat() local63 MS_EXCEPTION_IF_NULL(info_builder); in SetCastFormat()64 info_builder->SetInputsFormat({format}); in SetCastFormat()65 info_builder->SetOutputsFormat({format}); in SetCastFormat()66 AnfAlgo::SetSelectKernelBuildInfo(info_builder->Build(), cast_node.get()); in SetCastFormat()
46 Ptr<FontSourcedInfoBuilder> info_builder = in Subset() local50 font_info.Attach(info_builder->GetFontInfo()); in Subset()
62 Ptr<FontSourcedInfoBuilder> info_builder = in MergeFontInfos() local65 current_font_info.Attach(info_builder->GetFontInfo()); in MergeFontInfos()
352 auto info_builder = in SetCastAndWeightFormat() local354 MS_EXCEPTION_IF_NULL(info_builder); in SetCastAndWeightFormat()355 info_builder->SetInputsFormat({format}); in SetCastAndWeightFormat()356 info_builder->SetOutputsFormat({format}); in SetCastAndWeightFormat()357 AnfAlgo::SetSelectKernelBuildInfo(info_builder->Build(), kernel_node.get()); in SetCastAndWeightFormat()421 auto info_builder = in RefreshCastAndParamWeightFormat() local423 MS_EXCEPTION_IF_NULL(info_builder); in RefreshCastAndParamWeightFormat()424 info_builder->SetInputsFormat({format}); in RefreshCastAndParamWeightFormat()425 info_builder->SetOutputsFormat({format}); in RefreshCastAndParamWeightFormat()426 AnfAlgo::SetSelectKernelBuildInfo(info_builder->Build(), cast_node.get()); in RefreshCastAndParamWeightFormat()