Searched defs:kernel_info_list (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/ |
| D | aicpu_kernel_metadata.cc | 30 …cpuMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in AicpuMetadataInfo()
60 … std::vector<std::shared_ptr<KernelBuildInfo>> *kernel_info_list) { in AicpuMetadataInfoForSpecialNodes()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/ |
| D | bisheng_kernel_metadata.cc | 24 …BiShengMetadataInfo(const CNodePtr &kernel_node, std::vector<KernelBuildInfoPtr> *const kernel_inf… in BiShengMetadataInfo()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/stub/dynamic_shape/ |
| D | dynamic_shape_stub.cc | 22 …ostMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in HostMetadataInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/akg/ |
| D | akg_kernel_metadata.cc | 25 void AkgMetadataInfo(const CNodePtr &kernel_node, std::vector<KernelBuildInfoPtr> *kernel_info_list… in AkgMetadataInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/host/ |
| D | host_kernel_metadata.cc | 26 …ostMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in HostMetadataInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/rts/ |
| D | rt_kernel_info.cc | 57 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) { in GetRtKelInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/ |
| D | merge_cast_to_op.cc | 180 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in MergeCastToNextOp() local
252 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in MergeCastToPriorOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/ |
| D | hccl_kernel_metadata.cc | 74 …cclMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel… in HcclMetadataInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/ |
| D | kernel_select_ascend.cc | 63 … const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> &kernel_info_list) { in KernelSelectDebugString()
247 const CNodePtr &node, const std::vector<kernel::KernelBuildInfoPtr> &kernel_info_list, in CollectNotMatchMessage()
329 …MatchKernelInfo(const CNodePtr &kernel_node, const std::vector<kernel::KernelBuildInfoPtr> &kernel… in SetMatchKernelInfo()
660 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list{}; in SelectKernelInfoWithMsg() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion.cc | 135 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | kernel_info_setter.cc | 155 std::vector<std::shared_ptr<KernelBuildInfo>> kernel_info_list; in GetSupportedTypesStr() local
184 std::vector<std::shared_ptr<KernelBuildInfo>> kernel_info_list; in SelectAkgKernel() local
244 std::vector<std::shared_ptr<KernelBuildInfo>> kernel_info_list; in SelectCustomKernel() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/pre_activate/ascend/format_type/ |
| D | merge_cast_to_op_test.cc | 49 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) override { in Query()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/pre_activate/ascend/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion_test.cc | 46 …void Query(const CNodePtr &cnode, std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_in… in Query()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | kernel_select_cpu.cc | 543 std::vector<std::shared_ptr<KernelBuildInfo>> kernel_info_list; in UpdateCustomKernelBuildInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | framework_utils.cc | 492 std::vector<std::shared_ptr<KernelBuildInfo>> *const kernel_info_list) { in ParseMetadata()
|