Home
last modified time | relevance | path

Searched refs:SetSelectKernelBuildInfo (Results 1 – 25 of 90) sorted by relevance

1234

/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc78 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu1.get()); in TEST_F()
79 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu2.get()); in TEST_F()
92 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast.get()); in TEST_F()
153 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu1.get()); in TEST_F()
154 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu2.get()); in TEST_F()
155 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu3.get()); in TEST_F()
156 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu4.get()); in TEST_F()
157 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu5.get()); in TEST_F()
158 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu6.get()); in TEST_F()
171 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast.get()); in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dcommon_subexpression_elimination_test.cc115 AnfAlgo::SetSelectKernelBuildInfo( in TEST_F()
122 AnfAlgo::SetSelectKernelBuildInfo(nullptr, add2.get()); in TEST_F()
153 AnfAlgo::SetSelectKernelBuildInfo( in TEST_F()
157 AnfAlgo::SetSelectKernelBuildInfo( in TEST_F()
164 AnfAlgo::SetSelectKernelBuildInfo( in TEST_F()
168 AnfAlgo::SetSelectKernelBuildInfo( in TEST_F()
175 AnfAlgo::SetSelectKernelBuildInfo( in TEST_F()
179 AnfAlgo::SetSelectKernelBuildInfo( in TEST_F()
186 AnfAlgo::SetSelectKernelBuildInfo( in TEST_F()
190 AnfAlgo::SetSelectKernelBuildInfo( in TEST_F()
Deliminate_redundant_op_test.cc65 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), cnode.get()); in SelectKernel()
111 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), sub.get()); in TEST_F()
112 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), add.get()); in TEST_F()
178 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), sub.get()); in TEST_F()
179 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), add.get()); in TEST_F()
255 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), sub.get()); in TEST_F()
256 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), add.get()); in TEST_F()
Dallreduce_fusion_test.cc67 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), node.get()); in TEST_F()
108 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), node.get()); in TEST_F()
150 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), node.get()); in TEST_F()
206 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), node.get()); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dlayer_norm_beta_gamma_backprop_fusion_test.cc85 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast0.get()); in TEST_F()
86 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast1.get()); in TEST_F()
101 AnfAlgo::SetSelectKernelBuildInfo(builder2.Build(), layer.get()); in TEST_F()
144 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast0.get()); in TEST_F()
145 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast1.get()); in TEST_F()
160 AnfAlgo::SetSelectKernelBuildInfo(builder2.Build(), layer.get()); in TEST_F()
203 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast0.get()); in TEST_F()
204 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast1.get()); in TEST_F()
219 AnfAlgo::SetSelectKernelBuildInfo(builder2.Build(), layer.get()); in TEST_F()
256 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast.get()); in TEST_F()
[all …]
Dtranspose_transdata_fusion_test.cc63 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), cnode.get()); in SelectKernel()
72 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), cnode.get()); in SelectKernel()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dcheck_consistency_test.cc71 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), make_tuple.get()); in TEST_F()
84 AnfAlgo::SetSelectKernelBuildInfo(builder2.Build(), cast.get()); in TEST_F()
95 AnfAlgo::SetSelectKernelBuildInfo(builder3.Build(), para.get()); in TEST_F()
132 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), make_tuple.get()); in TEST_F()
145 AnfAlgo::SetSelectKernelBuildInfo(builder2.Build(), cast.get()); in TEST_F()
156 AnfAlgo::SetSelectKernelBuildInfo(builder3.Build(), para.get()); in TEST_F()
Dinsert_cast_test.cc79 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), node.get()); in TEST_F()
82 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), node.get()); in TEST_F()
133 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), node.get()); in TEST_F()
136 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), node.get()); in TEST_F()
Dmerge_cast_to_op_test.cc101 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), four2five.get()); in TEST_F()
114 AnfAlgo::SetSelectKernelBuildInfo(builder2.Build(), cast.get()); in TEST_F()
164 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast.get()); in TEST_F()
177 AnfAlgo::SetSelectKernelBuildInfo(builder2.Build(), five2four.get()); in TEST_F()
Dinsert_trans_op_test.cc58 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), add.get()); in GetSingleOutputGraph()
82 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), max_pool.get()); in GetMutilpleOutputGraph()
101 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), cnode.get()); in SelectKernel()
Dremove_internal_output_test.cc61 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), add.get()); in GetSingleOutputGraph()
90 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), max_pool.get()); in GetMutilpleOutputGraph()
108 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), cnode.get()); in SelectKernel()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc72 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu1.get()); in TEST_F()
115 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), conv2d_backprop_filter.get()); in TEST_F()
173 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), dynamic_rnn.get()); in TEST_F()
227 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), layer_norm.get()); in TEST_F()
273 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu1.get()); in TEST_F()
274 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), relu2.get()); in TEST_F()
287 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), cast.get()); in TEST_F()
344 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), add.get()); in TEST_F()
346 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), conv2d.get()); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dbn_grad_split_test.cc78 AnfAlgo::SetSelectKernelBuildInfo(builder2.Build(), param1.get()); in TEST_F()
95 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), bn_grad.get()); in TEST_F()
144 AnfAlgo::SetSelectKernelBuildInfo(builder2.Build(), param1.get()); in TEST_F()
157 AnfAlgo::SetSelectKernelBuildInfo(builder1.Build(), bn_grad.get()); in TEST_F()
Dtransdata_split_test.cc56 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), cnode.get()); in SelectKernel()
65 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), cnode.get()); in SelectKernel()
84 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), cnode.get()); in SelectKernel()
93 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), cnode.get()); in SelectKernel()
Dbn_split_test.cc87 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), bn.get()); in TEST_F()
136 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), bn.get()); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dtrans_op_format_refine.cc44 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), node.get()); in Process()
50 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), node.get()); in Process()
Ddynamic_rnn_grad_reformat.cc58 AnfAlgo::SetSelectKernelBuildInfo(matmul_new_builder->Build(), matmul.get()); in Process()
72 AnfAlgo::SetSelectKernelBuildInfo(split_new_builder->Build(), split_v.get()); in Process()
Drectify_do_mask_kernel_info.cc118 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), do_mask.get()); in RectifyDropOutDoMaskKernelInfo()
140 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), node.get()); in RectifyKernelInfoInPynativeProcess()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Ddecrease_transfer_precision.cc175 AnfAlgo::SetSelectKernelBuildInfo(info_1, cnode.get()); in Process_Father()
198 AnfAlgo::SetSelectKernelBuildInfo(gk_builder_info->Build(), node.get()); in Process_Father()
230 AnfAlgo::SetSelectKernelBuildInfo(gk_builder_info->Build(), node.get()); in Process_Father()
252 AnfAlgo::SetSelectKernelBuildInfo(gk_builder_info->Build(), node.get()); in Process_Son()
288 AnfAlgo::SetSelectKernelBuildInfo(info_1, cnode.get()); in Process_Son()
Dsubstitute_dropout.cc66 …AnfAlgo::SetSelectKernelBuildInfo(uniform_real_kernel_info_builder->Build(), uniform_real_node.get… in PreProcess()
78 AnfAlgo::SetSelectKernelBuildInfo(dropout_kernel_info_builder->Build(), new_dropout_node.get()); in PreProcess()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_ascend.cc357 AnfAlgo::SetSelectKernelBuildInfo(info_builder->Build(), kernel_node.get()); in SetCastAndWeightFormat()
391 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), real_input_node.get()); in SetWeightFormat()
398 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), real_input_node.get()); in SetWeightFormat()
426 AnfAlgo::SetSelectKernelBuildInfo(info_builder->Build(), cast_node.get()); in RefreshCastAndParamWeightFormat()
486 AnfAlgo::SetSelectKernelBuildInfo(selected_kernel_info, kernel_node.get()); in SetMatchedKernelInfo()
529 AnfAlgo::SetSelectKernelBuildInfo(selected_kernel_info, kernel_node.get()); in SelectKernelInfo()
594 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), kernel_node.get()); in SetKernelInfo()
Dkernel_select_graph_kernel.cc74 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), input_kernel_node.get()); in ResetKernelBuildInfo()
97 AnfAlgo::SetSelectKernelBuildInfo(selected_kernel_info_ptr, cnode.get()); in UpdateKernelInfo()
301 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), input_list[i].get()); in UpdateInputsKernelInfo()
421 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), input_list[i].get()); in UpdateFormatsAndDtypes()
442 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), input_node.get()); in UpdateFormatsAndDtypes()
477 AnfAlgo::SetSelectKernelBuildInfo(graph_selected_info, kernel_node.get()); in SetGraphKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dtransposed_update_fusion.cc61 AnfAlgo::SetSelectKernelBuildInfo(op_builder.Build(), perm_const.get()); in CreatePermValueNode()
100 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), transpose.get()); in Process()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc166 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), real_input_node.get()); in SetTensorDeviceInfo()
181 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), real_input_node.get()); in SetTensorDeviceInfo()
303 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), input_list[i].get()); in SetGraphKernelInfo()
340 AnfAlgo::SetSelectKernelBuildInfo(graph_selected_info, kernel_node.get()); in SetGraphKernelInfo()
450 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), kernel_node.get()); in SetKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Drelu_v2_pass.cc91 AnfAlgo::SetSelectKernelBuildInfo(build_info, new_node.get()); in CreateReluV2()
119 AnfAlgo::SetSelectKernelBuildInfo(build_info, new_node.get()); in CreateReluGradV2()

1234