| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/ |
| D | transdata_split_test.cc | 51 builder.SetInputsDeviceType({kFloat16->type_id()}); in SelectKernel()
60 builder.SetInputsDeviceType({kFloat16->type_id()}); in SelectKernel()
79 builder.SetInputsDeviceType({kFloat16->type_id()}); in SelectKernel()
88 builder.SetInputsDeviceType({kFloat16->type_id()}); in SelectKernel()
122 builder.SetInputsDeviceType({kFloat16->type_id()}); in TEST_F()
170 builder.SetInputsDeviceType({kFloat16->type_id()}); in TEST_F()
|
| D | bn_split_test.cc | 82 builder.SetInputsDeviceType( in TEST_F()
131 builder.SetInputsDeviceType( in TEST_F()
|
| D | bn_grad_split_test.cc | 88 builder1.SetInputsDeviceType( in TEST_F()
152 builder1.SetInputsDeviceType( in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion_test.cc | 49 …builder.SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16, kNumberTypeFloat16, kNumberTy… in Query()
79 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F()
98 …builder2.SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16, kNumberTypeFloat16, kNumberT… in TEST_F()
140 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F()
157 …builder2.SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16, kNumberTypeFloat16, kNumberT… in TEST_F()
199 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F()
216 …builder2.SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16, kNumberTypeFloat16, kNumberT… in TEST_F()
253 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F()
265 …builder2.SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16, kNumberTypeFloat16, kNumberT… in TEST_F()
307 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F()
[all …]
|
| D | transpose_transdata_fusion_test.cc | 58 builder.SetInputsDeviceType({kFloat16->type_id()}); in SelectKernel()
67 builder.SetInputsDeviceType({kFloat16->type_id()}); in SelectKernel()
103 builder.SetInputsDeviceType({kFloat16->type_id()}); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/ |
| D | merge_cast_to_op_test.cc | 53 builder.SetInputsDeviceType({kNumberTypeFloat32}); in Query()
60 builder.SetInputsDeviceType({kNumberTypeFloat16}); in Query()
99 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F()
112 builder2.SetInputsDeviceType({kNumberTypeFloat32}); in TEST_F()
162 builder1.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F()
175 builder2.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F()
|
| D | check_consistency_test.cc | 69 builder1.SetInputsDeviceType({kNumberTypeFloat32}); in TEST_F()
82 builder2.SetInputsDeviceType({kNumberTypeFloat32}); in TEST_F()
130 builder1.SetInputsDeviceType({kNumberTypeFloat32}); in TEST_F()
143 builder2.SetInputsDeviceType({kNumberTypeFloat16}); in TEST_F()
|
| D | insert_cast_test.cc | 59 builder.SetInputsDeviceType({kFloat16->type_id(), kFloat16->type_id()}); in TEST_F()
66 builder1.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
116 builder.SetInputsDeviceType({kFloat16->type_id()}); in TEST_F()
120 builder1.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
|
| D | insert_trans_op_test.cc | 52 builder.SetInputsDeviceType({kFloat16->type_id(), kFloat16->type_id()}); in GetSingleOutputGraph()
78 builder.SetInputsDeviceType({kFloat16->type_id()}); in GetMutilpleOutputGraph()
98 builder.SetInputsDeviceType({kFloat16->type_id()}); in SelectKernel()
|
| D | remove_internal_output_test.cc | 55 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id()}); in GetSingleOutputGraph()
86 builder.SetInputsDeviceType({kFloat32->type_id()}); in GetMutilpleOutputGraph()
103 builder.SetInputsDeviceType({kFloat16->type_id()}); in SelectKernel()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/ |
| D | buffer_fusion_test.cc | 69 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
84 builder1.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
140 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
163 builder1.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
176 builder2.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id()}); in TEST_F()
230 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
253 builder1.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
266 builder2.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
312 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
324 builder2.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id()}); in TEST_F()
[all …]
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/pass/ |
| D | allreduce_fusion_test.cc | 55 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
96 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
137 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
194 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
|
| D | eliminate_redundant_op_test.cc | 62 builder.SetInputsDeviceType({kFloat16->type_id()}); in SelectKernel()
105 builder.SetInputsDeviceType({kFloat16->type_id(), kFloat16->type_id()}); in TEST_F()
172 builder.SetInputsDeviceType({kFloat16->type_id(), kFloat16->type_id()}); in TEST_F()
249 builder.SetInputsDeviceType({kFloat16->type_id(), kFloat16->type_id()}); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/tbe/ |
| D | tbe_json_creator_test.cc | 64 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
107 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id()}); in TEST_F()
163 …builder.SetInputsDeviceType({kFloat16->type_id(), kFloat16->type_id(), kFloat16->type_id(), kFloat… in TEST_F()
219 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id(), kFloat32->type_id()}); in TEST_F()
264 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
279 builder1.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
336 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id()}); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | dynamic_rnn_grad_reformat.cc | 53 matmul_new_builder->SetInputsDeviceType({kNumberTypeFloat16, kNumberTypeFloat16}); in Process()
67 split_new_builder->SetInputsDeviceType(split_kernel_build_info->GetAllInputDeviceTypes()); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | substitute_dropout.cc | 61 uniform_real_kernel_info_builder->SetInputsDeviceType({kNumberTypeInt32}); in PreProcess()
77 …dropout_kernel_info_builder->SetInputsDeviceType({old_kernel_info->GetInputDeviceType(0), kNumberT… in PreProcess()
|
| D | decrease_transfer_precision.cc | 168 graph_info_builder.SetInputsDeviceType(cnode_input_type); in Process_Father()
251 gk_builder_info->SetInputsDeviceType(new_inputs_type); in Process_Son()
281 node_info_builder.SetInputsDeviceType(cnode_input_type); in Process_Son()
|
| D | decrease_compute_precision.cc | 45 builder.SetInputsDeviceType({input_type}); in AddCastCNode()
183 graph_info_builder.SetInputsDeviceType(cnode_input_type); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | host_kernel_metadata.cc | 57 builder.SetInputsDeviceType(inputs_type); in HostMetadataInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | transpose_transdata_fusion.cc | 50 new_transdata_builder->SetInputsDeviceType(transdata_kernel_build_info->GetAllInputDeviceTypes()); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_metadata.cc | 73 builder.SetInputsDeviceType(inputs_type); in AicpuMetadataInfoForSpecialNodes()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | rt_kernel_info.cc | 77 kernel_build_info_builder->SetInputsDeviceType(input_types); in GetRtKelInfo()
|
| D | label_switch.cc | 86 builder.SetInputsDeviceType({input_type[i]}); in GetKernelInfo()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/ |
| D | mem_reuse_test.cc | 95 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id()}); in CreateKernelGraph()
126 relu_builder.SetInputsDeviceType({kFloat32->type_id()}); in CreateKernelGraph()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | add_relu_v2_fusion.cc | 47 builder.SetInputsDeviceType(inputs_type); in GenerateKernelBuildInfo()
|