Home
last modified time | relevance | path

Searched defs:kernel_info_list (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_query.cc32std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) { in FilterInvalidKernelInfo()
71 void CheckKernelInfoListEmpty(const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_i… in CheckKernelInfoListEmpty()
80 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) { in KernelQueryAll()
98 …KernelQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *ke… in KernelQuery()
128 … AICPUQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *ke… in AICPUQuery()
139 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in IsSupportedByAICPU() local
153 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in IsSupportedByAICore() local
Dcommon_utils.cc430 std::vector<std::shared_ptr<KernelBuildInfo>> *const kernel_info_list) { in ParseMetadata()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc27 …cpuMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in AicpuMetadataInfo()
52std::vector<std::shared_ptr<KernelBuildInfo>> *kernel_info_list) { in AicpuMetadataInfoForSpecialNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.h60 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) { in Query()
82std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) { in GetTbeKernelMetaInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_ascend.cc204 …const CNodePtr &kernel_node, const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> &kernel_i… in ChooseMatchedKernelInfo()
224 …const CNodePtr &cnode, const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> &kernel_info_li… in FilteredKernelInfoByDtype()
290 …const CNodePtr &cnode, const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> &kernel_info_li… in FilterRaisedOrReducePrecisionMatchedKernelInfo()
457const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> &kernel_info_list) { in SetMatchedKernelInfo()
496 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in SelectKernelInfo() local
568 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in SetKernelInfo() local
Dkernel_select_graph_kernel.cc87 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in UpdateKernelInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_metadata.cc25 std::vector<std::shared_ptr<KernelBuildInfo>> *const kernel_info_list) { in AkgMetadataInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_metadata.cc30 …ostMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in HostMetadataInfo()
/third_party/mindspore/tests/ut/cpp/stub/dynamic_shape/
Ddynamic_shape_stub.cc46 …ostMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in HostMetadataInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Drt_kernel_info.cc54 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) { in GetRtKelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel_metadata.cc65 …cclMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in HcclMetadataInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dmerge_cast_to_op.cc164 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in MergeCastToNextOp() local
234 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in MergeCastToPriorOp() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dtransposed_update_fusion.cc86 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in Process() local
Dlayer_norm_beta_gamma_backprop_fusion.cc131 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in Process() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc49 …TbeMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in TbeMetadataInfo()
54 …elect::TbeKernelSelect(CNodePtr kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in TbeKernelSelect()
196 std::vector<std::shared_ptr<KernelBuildInfo>> kernel_info_list; in FilterInVaildKernelInfo() local
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dmerge_cast_to_op_test.cc47 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) override { in Query()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc79 std::vector<std::shared_ptr<KernelBuildInfo>> kernel_info_list; in SupportedTypeList() local
106 std::vector<std::shared_ptr<KernelBuildInfo>> kernel_info_list; in SelectAkgKernel() local
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dlayer_norm_beta_gamma_backprop_fusion_test.cc44 …void Query(const CNodePtr &cnode, std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_in… in Query()