Home
last modified time | relevance | path

Searched refs:KernelBuildInfoBuilder (Results 1 – 25 of 96) sorted by relevance

1234

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_build_info.cc146 void KernelBuildInfo::KernelBuildInfoBuilder::SetKernelType(const KernelType &kernel_type) { in SetKernelType()
151 void KernelBuildInfo::KernelBuildInfoBuilder::SetOriginDataFormat(const std::string &origin_data_fo… in SetOriginDataFormat()
156 void KernelBuildInfo::KernelBuildInfoBuilder::SetInputsFormat(const std::vector<std::string> &input… in SetInputsFormat()
161 void KernelBuildInfo::KernelBuildInfoBuilder::SetOutputsFormat(const std::vector<std::string> &outp… in SetOutputsFormat()
166 void KernelBuildInfo::KernelBuildInfoBuilder::SetInputsDeviceType(const std::vector<TypeId> &inputs… in SetInputsDeviceType()
171 void KernelBuildInfo::KernelBuildInfoBuilder::SetOutputsDeviceType(const std::vector<TypeId> &outpu… in SetOutputsDeviceType()
176 void KernelBuildInfo::KernelBuildInfoBuilder::SetFusionType(FusionType fusion_type) { in SetFusionType()
181 void KernelBuildInfo::KernelBuildInfoBuilder::SetOutputDataDesc(const std::vector<nlohmann::json> &… in SetOutputDataDesc()
186 void KernelBuildInfo::KernelBuildInfoBuilder::SetProcessor(Processor processor) { in SetProcessor()
191 std::shared_ptr<KernelBuildInfo> KernelBuildInfo::KernelBuildInfoBuilder::Build() { return kernel_b… in Build()
[all …]
Dkernel_build_info.h32 class KernelBuildInfoBuilder;
124 class KernelBuildInfo::KernelBuildInfoBuilder {
126 KernelBuildInfoBuilder() { kernel_build_info_ = std::make_shared<KernelBuildInfo>(); } in KernelBuildInfoBuilder() function
128 explicit KernelBuildInfoBuilder(std::shared_ptr<KernelBuildInfo> kernel_build_info) in KernelBuildInfoBuilder() function
147 ~KernelBuildInfoBuilder() = default;
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dtransdata_split_test.cc34 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
49 KernelBuildInfoBuilder builder; in SelectKernel()
58 KernelBuildInfoBuilder builder; in SelectKernel()
77 KernelBuildInfoBuilder builder; in SelectKernel()
86 KernelBuildInfoBuilder builder; in SelectKernel()
120 KernelBuildInfoBuilder builder; in TEST_F()
168 KernelBuildInfoBuilder builder; in TEST_F()
Dbn_grad_split_test.cc75 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
81 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
141 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
147 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dcheck_consistency_test.cc66 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
79 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
92 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder3; in TEST_F()
127 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
140 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
153 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder3; in TEST_F()
Dinsert_trans_op_test.cc34 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
50 KernelBuildInfoBuilder builder; in GetSingleOutputGraph()
74 KernelBuildInfoBuilder builder; in GetMutilpleOutputGraph()
94 KernelBuildInfoBuilder builder; in SelectKernel()
Dmerge_cast_to_op_test.cc50 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in Query()
57 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in Query()
96 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
109 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
159 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
172 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
Dremove_internal_output_test.cc29 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
53 KernelBuildInfoBuilder builder; in GetSingleOutputGraph()
82 KernelBuildInfoBuilder builder; in GetMutilpleOutputGraph()
101 KernelBuildInfoBuilder builder; in SelectKernel()
Dinsert_cast_test.cc56 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
64 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
113 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
118 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc40 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
66 KernelBuildInfoBuilder builder; in TEST_F()
81 KernelBuildInfoBuilder builder1; in TEST_F()
137 KernelBuildInfoBuilder builder; in TEST_F()
160 KernelBuildInfoBuilder builder1; in TEST_F()
173 KernelBuildInfoBuilder builder2; in TEST_F()
227 KernelBuildInfoBuilder builder; in TEST_F()
250 KernelBuildInfoBuilder builder1; in TEST_F()
263 KernelBuildInfoBuilder builder2; in TEST_F()
309 KernelBuildInfoBuilder builder; in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dtranspose_transdata_fusion_test.cc32 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
56 KernelBuildInfoBuilder builder; in SelectKernel()
65 KernelBuildInfoBuilder builder; in SelectKernel()
99 KernelBuildInfoBuilder builder; in TEST_F()
Dlayer_norm_beta_gamma_backprop_fusion_test.cc46 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in Query()
76 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
95 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
137 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
154 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
196 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
213 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
250 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
262 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
304 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/enhancer/
Dinsert_tensor_move_for_getnext_test.cc31 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
48 KernelBuildInfoBuilder builder; in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Deliminate_redundant_op_test.cc43 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
58 KernelBuildInfoBuilder builder; in SelectKernel()
102 KernelBuildInfoBuilder builder; in TEST_F()
169 KernelBuildInfoBuilder builder; in TEST_F()
246 KernelBuildInfoBuilder builder; in TEST_F()
Dallreduce_fusion_test.cc52 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
93 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
134 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
191 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc31 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
61 KernelBuildInfoBuilder builder; in TEST_F()
104 KernelBuildInfoBuilder builder; in TEST_F()
160 KernelBuildInfoBuilder builder; in TEST_F()
216 KernelBuildInfoBuilder builder; in TEST_F()
261 KernelBuildInfoBuilder builder; in TEST_F()
276 KernelBuildInfoBuilder builder1; in TEST_F()
333 KernelBuildInfoBuilder builder; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Drefresh_parameter_format.cc32 std::shared_ptr<kernel::KernelBuildInfo::KernelBuildInfoBuilder> builder = in DoRefresh()
33 std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(); in DoRefresh()
Dtransposed_update_fusion.cc58 kernel::KernelBuildInfo::KernelBuildInfoBuilder op_builder; in CreatePermValueNode()
95 auto builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(ori_build_info); in Process()
/third_party/mindspore/tests/ut/cpp/session/
Dkernel_graph_test.cc28 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
47 std::shared_ptr<KernelBuildInfoBuilder> builder = std::make_shared<KernelBuildInfoBuilder>(); in TEST_F()
Danf_runtime_algorithm_test.cc29 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
255 KernelBuildInfoBuilder builder; in TEST_F()
274 KernelBuildInfoBuilder builder; in TEST_F()
293 KernelBuildInfoBuilder builder; in TEST_F()
373 KernelBuildInfoBuilder builder; in TEST_F()
406 KernelBuildInfoBuilder builder; in TEST_F()
460 KernelBuildInfoBuilder builder; in TEST_F()
477 KernelBuildInfoBuilder builder; in TEST_F()
495 KernelBuildInfoBuilder builder; in TEST_F()
646 KernelBuildInfoBuilder builder; in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Ddecrease_transfer_precision.cc166 kernel::KernelBuildInfo::KernelBuildInfoBuilder graph_info_builder; in Process_Father()
195 …std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInf… in Process_Father()
222 …auto gk_builder_info = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(old_build… in Process_Father()
247 …std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInf… in Process_Son()
279 kernel::KernelBuildInfo::KernelBuildInfoBuilder node_info_builder; in Process_Son()
Dsubstitute_dropout.cc59 …orm_real_kernel_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(); in PreProcess()
75 …_kernel_info_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(old_kerne… in PreProcess()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Ddynamic_rnn_grad_reformat.cc50 auto matmul_new_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(); in Process()
64 auto split_new_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(); in Process()
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc38 using KernelBuildInfoBuilder = kernel::KernelBuildInfo::KernelBuildInfoBuilder; typedef
93 KernelBuildInfoBuilder builder; in CreateKernelGraph()
123 KernelBuildInfoBuilder relu_builder; in CreateKernelGraph()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc83 kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder; in CreateSendApplyKernel()
103 kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder; in CreateRecvApplyKernel()
560 kernel::KernelBuildInfo::KernelBuildInfoBuilder KernelAdjust::CreateMngKernelBuilder( in CreateMngKernelBuilder()
562 kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder; in CreateMngKernelBuilder()
575 kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder = CreateMngKernelBuilder( in CreateStreamSwitchOp()
608 kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder = CreateMngKernelBuilder( in CreateStreamActiveOp()
641 kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder; in CreateEndOfSequenceOP()
675 kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder = CreateMngKernelBuilder( in CreateStreamAssignAddnOP()
903 kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder; in CreateNPUGetFloatStatus()
926 kernel::KernelBuildInfo::KernelBuildInfoBuilder selected_kernel_builder; in CreateNPUClearStatus()
[all …]

1234