Searched refs:KernelQuery (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.h | 55 class KernelQuery {
57 KernelQuery() = default;
58 virtual ~KernelQuery() = default;
61 kernel::KernelQuery(kernel_node, kernel_info_list); in Query()
75 using KernelQueryPtr = std::shared_ptr<KernelQuery>;
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | merge_cast_to_op.h | 30 …tternProcessPass("merge_cast_to_op", multigraph), kernel_query_(std::make_shared<KernelQuery>()) {}
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion.h | 30 kernel_query_(std::make_shared<KernelQuery>()) {}
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_query.h | 28 void KernelQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>>…
|
| D | kernel_query.cc | 98 void KernelQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>>… in KernelQuery() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | insert_tensor_move_for_hccl_op.h | 29 kernel_query_(std::make_shared<KernelQuery>()) {}
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/enhancer/ |
| D | insert_tensor_move_for_hccl_op_test.cc | 42 class MockInsertTensorMoveForHcclKernelQuery : public KernelQuery {
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/ |
| D | merge_cast_to_op_test.cc | 42 class MockMergeCastToOpKernelQuery : public KernelQuery {
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_ascend.cc | 505 kernel::KernelQuery(kernel_node, &kernel_info_list, kernel_type); in SelectKernelInfo()
570 kernel::KernelQuery(kernel_node, &kernel_info_list, kernel_type); in SetKernelInfo()
|
| D | kernel_select_graph_kernel.cc | 88 kernel::KernelQuery(cnode, &kernel_info_list, KernelType::AKG_KERNEL); in UpdateKernelInfo()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion_test.cc | 40 class MockLayerNormBetaGammaBackpropFusionKernelQuery : public KernelQuery {
|