Searched refs:kernel_build_info_builder (Results 1 – 7 of 7) sorted by relevance
68 …auto kernel_build_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>… in GetRtKelInfo() local71 MS_EXCEPTION_IF_NULL(kernel_build_info_builder); in GetRtKelInfo()72 …kernel_build_info_builder->SetInputsFormat(std::vector<std::string>(input_num, kOpFormat_DEFAULT)); in GetRtKelInfo()77 kernel_build_info_builder->SetInputsDeviceType(input_types); in GetRtKelInfo()79 kernel_build_info_builder->SetOutputsFormat({kOpFormat_DEFAULT}); in GetRtKelInfo()80 kernel_build_info_builder->SetOutputsDeviceType({TypeId::kObjectTypeUMonad}); in GetRtKelInfo()82 kernel_build_info_builder->SetFusionType(kernel::FusionType::OPAQUE); in GetRtKelInfo()83 kernel_build_info_builder->SetProcessor(kernel::Processor::AICORE); in GetRtKelInfo()84 kernel_build_info_builder->SetKernelType(KernelType::RT_KERNEL); in GetRtKelInfo()85 kernel_info_list->push_back(kernel_build_info_builder->Build()); in GetRtKelInfo()
128 …auto kernel_build_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>… in CreateScalarTensorValueNode() local129 kernel_build_info_builder->SetOutputsFormat(std::vector<std::string>{info.format}); in CreateScalarTensorValueNode()131 kernel_build_info_builder->SetOutputsDeviceType(types); in CreateScalarTensorValueNode()132 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), new_value_node.get()); in CreateScalarTensorValueNode()
142 …auto kernel_build_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>… in ConvertToScalarTensor() local143 kernel_build_info_builder->SetOutputsFormat(std::vector<std::string>{GetFormat(value_node)}); in ConvertToScalarTensor()144 kernel_build_info_builder->SetOutputsDeviceType(std::vector<TypeId>{type_id}); in ConvertToScalarTensor()145 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), new_value_node.get()); in ConvertToScalarTensor()
389 …auto kernel_build_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>… in Process() local390 kernel_build_info_builder->SetOutputsFormat(std::vector<std::string>{kOpFormat_DEFAULT}); in Process()391 kernel_build_info_builder->SetOutputsDeviceType(std::vector<TypeId>{kNumberTypeUInt8}); in Process()392 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), mask_input.get()); in Process()
904 …auto kernel_build_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>… in MakeValueNode() local905 MS_EXCEPTION_IF_NULL(kernel_build_info_builder); in MakeValueNode()907 kernel_build_info_builder->SetOutputsFormat(std::vector<std::string>{kOpFormat_DEFAULT}); in MakeValueNode()914 kernel_build_info_builder->SetOutputsDeviceType(types); in MakeValueNode()915 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), new_value_node.get()); in MakeValueNode()
528 …auto kernel_build_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>… in SetKernelInfoForNode() local529 MS_EXCEPTION_IF_NULL(kernel_build_info_builder); in SetKernelInfoForNode()547 kernel_build_info_builder->SetOutputsFormat(formats); in SetKernelInfoForNode()548 kernel_build_info_builder->SetOutputsDeviceType(types); in SetKernelInfoForNode()549 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), node.get()); in SetKernelInfoForNode()
313 …auto kernel_build_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>… in ConstructRunOpParameter() local317 kernel_build_info_builder->SetOutputsFormat(std::vector<std::string>{kOpFormat_DEFAULT}); in ConstructRunOpParameter()319 kernel_build_info_builder->SetOutputsDeviceType(std::vector<TypeId>{param_init_data_type}); in ConstructRunOpParameter()321 kernel_build_info_builder->SetOutputsFormat(std::vector<std::string>{device_address->format()}); in ConstructRunOpParameter()322 kernel_build_info_builder->SetOutputsDeviceType(std::vector<TypeId>{device_address->type_id()}); in ConstructRunOpParameter()323 kernel_build_info_builder->SetOutputsReshapeType({input_tensor->padding_type()}); in ConstructRunOpParameter()326 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), param.get()); in ConstructRunOpParameter()1092 …auto kernel_build_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>… in CreateValueNodeKernelGraph() local1093 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), new_value_node.get()); in CreateValueNodeKernelGraph()