Home
last modified time | relevance | path

Searched refs:GetSelectKernelBuildInfo (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Drectify_do_mask_kernel_info.cc114 …hared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(do_mask)); in RectifyDropOutDoMaskKernelInfo()
136 …e_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(node)); in RectifyKernelInfoInPynativeProcess()
153 auto ori_build_info = AnfAlgo::GetSelectKernelBuildInfo(out_node); in ReSelecChildNodeKernelInfo()
155 auto new_build_info = AnfAlgo::GetSelectKernelBuildInfo(out_node); in ReSelecChildNodeKernelInfo()
Ddynamic_rnn_grad_reformat.cc48 auto matmul_kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(matmul); in Process()
62 auto split_kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(split_v); in Process()
Dconvert_unsupported_transnode_to_aicpu.cc40 auto kernel_builder_info = AnfAlgo::GetSelectKernelBuildInfo(node); in Process()
Dtrans_op_format_refine.cc39 …e_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(node)); in Process()
Dmerge_cast_to_op.cc176 auto ori_kernel_info = AnfAlgo::GetSelectKernelBuildInfo(next_node); in MergeCastToNextOp()
249 auto ori_kernel_info = AnfAlgo::GetSelectKernelBuildInfo(prior_op); in MergeCastToPriorOp()
Dconvert_cast_format.cc62 …hared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(cast_node… in SetCastFormat()
Ddeal_ref_and_split_unsupported_transdata.cc276 auto kernel_info = AnfAlgo::GetSelectKernelBuildInfo(cnode); in SplitTransdataIfNotSupported()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dtranspose_transdata_fusion.cc40 auto transpose_kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(transpose_cnode); in Process()
41 auto transdata_kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(transdata_cnode); in Process()
Dtransposed_update_fusion.cc93 auto ori_build_info = AnfAlgo::GetSelectKernelBuildInfo(transpose); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Ddecrease_transfer_precision.cc195 …e_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(node)); in Process_Father()
221 auto old_builder_info = AnfAlgo::GetSelectKernelBuildInfo(node); in Process_Father()
247 …e_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(node)); in Process_Son()
Dsubstitute_dropout.cc74 auto old_kernel_info = AnfAlgo::GetSelectKernelBuildInfo(node); in PreProcess()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_inference_session.cc161 auto kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(parameter); in CompareInput()
204 … data_type_to_string(AnfAlgo::GetSelectKernelBuildInfo(para)->GetOutputDeviceType(0)) + " }"; in InputsInfo()
Dgpu_inference_session.cc161 auto kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(parameter); in CompareInput()
204 … data_type_to_string(AnfAlgo::GetSelectKernelBuildInfo(para)->GetOutputDeviceType(0)) + " }"; in InputsInfo()
Danf_runtime_algorithm.h230 static kernel::KernelBuildInfoPtr GetSelectKernelBuildInfo(const AnfNodePtr &node);
Danf_runtime_algorithm.cc1322 …e_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(node)); in SetFusionType()
1331 …e_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(node)); in SetOutputDataDesc()
1379 KernelBuildInfoPtr AnfRuntimeAlgorithm::GetSelectKernelBuildInfo(const AnfNodePtr &node) { in GetSelectKernelBuildInfo() function in mindspore::session::AnfRuntimeAlgorithm
Dsession_basic.cc1775 auto kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(parameter); in GetModelInputsInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_ascend.cc353 …hared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(kernel_no… in SetCastAndWeightFormat()
384 auto selected_kernel_info = AnfAlgo::GetSelectKernelBuildInfo(kernel_node); in SetWeightFormat()
422 …hared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInfo(input_nod… in RefreshCastAndParamWeightFormat()
434 auto selected_kernel_info = AnfAlgo::GetSelectKernelBuildInfo(kernel_node); in SetTensorDeviceInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcudnn_inplace_fusion.cc148 auto build_info = AnfAlgo::GetSelectKernelBuildInfo(src); in CopyKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc246 auto ori_build_info = AnfAlgo::GetSelectKernelBuildInfo(trans_data); in RefreshKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc270 auto kernel_build_info_tmp = AnfAlgo::GetSelectKernelBuildInfo(cnode_ptr_); in TbeCheckSupported()