Home
last modified time | relevance | path

Searched refs:SetOutputsFormat (Results 1 – 25 of 84) sorted by relevance

1234

/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dcheck_consistency_test.cc68 builder1.SetOutputsFormat({"NCHW"}); in TEST_F()
81 builder2.SetOutputsFormat({"NCHW"}); in TEST_F()
93 builder3.SetOutputsFormat({"NCHW"}); in TEST_F()
129 builder1.SetOutputsFormat({"NCHW"}); in TEST_F()
142 builder2.SetOutputsFormat({"NCHW"}); in TEST_F()
154 builder3.SetOutputsFormat({"NCHW"}); in TEST_F()
Dmerge_cast_to_op_test.cc52 builder.SetOutputsFormat({kOpFormat_NC1HWC0}); in Query()
59 builder.SetOutputsFormat({kOpFormat_NCHW}); in Query()
98 builder1.SetOutputsFormat({kOpFormat_NC1HWC0}); in TEST_F()
111 builder2.SetOutputsFormat({kOpFormat_NCHW}); in TEST_F()
161 builder1.SetOutputsFormat({kOpFormat_NCHW}); in TEST_F()
174 builder2.SetOutputsFormat({kOpFormat_NCHW}); in TEST_F()
Dinsert_cast_test.cc58 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
67 builder1.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
115 builder.SetOutputsFormat({"NC1HWC0", "NC1HWC0"}); in TEST_F()
121 builder1.SetOutputsFormat({"DefaultFormat"}); in TEST_F()
Dinsert_trans_op_test.cc55 builder.SetOutputsFormat({format}); in GetSingleOutputGraph()
79 builder.SetOutputsFormat({format, format}); in GetMutilpleOutputGraph()
99 builder.SetOutputsFormat({"NC1HWC0"}); in SelectKernel()
Dremove_internal_output_test.cc58 builder.SetOutputsFormat({kOpFormat_NC1HWC0}); in GetSingleOutputGraph()
87 builder.SetOutputsFormat({kOpFormat_NC1HWC0, kOpFormat_NC1HWC0}); in GetMutilpleOutputGraph()
104 builder.SetOutputsFormat({kOpFormat_DEFAULT}); in SelectKernel()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dtransdata_split_test.cc52 builder.SetOutputsFormat({"NC1HWC0"}); in SelectKernel()
61 builder.SetOutputsFormat({"NC1HWC0"}); in SelectKernel()
80 builder.SetOutputsFormat({"NCHW"}); in SelectKernel()
89 builder.SetOutputsFormat({"NCHW"}); in SelectKernel()
123 builder.SetOutputsFormat({kOpFormat_C1HWNCoC0}); in TEST_F()
171 builder.SetOutputsFormat({"NCHW"}); in TEST_F()
Dbn_grad_split_test.cc76 builder2.SetOutputsFormat({kOpFormat_NC1HWC0}); in TEST_F()
85 builder1.SetOutputsFormat( in TEST_F()
142 builder2.SetOutputsFormat({kOpFormat_NC1HWC0}); in TEST_F()
150 builder1.SetOutputsFormat( in TEST_F()
Dbn_split_test.cc80 builder.SetOutputsFormat( in TEST_F()
129 builder.SetOutputsFormat( in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dlayer_norm_beta_gamma_backprop_fusion_test.cc48 builder.SetOutputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT}); in Query()
78 builder1.SetOutputsFormat({kOpFormat_DEFAULT}); in TEST_F()
97 builder2.SetOutputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT}); in TEST_F()
139 builder1.SetOutputsFormat({kOpFormat_DEFAULT}); in TEST_F()
156 builder2.SetOutputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT}); in TEST_F()
198 builder1.SetOutputsFormat({kOpFormat_DEFAULT}); in TEST_F()
215 builder2.SetOutputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT}); in TEST_F()
252 builder1.SetOutputsFormat({kOpFormat_DEFAULT}); in TEST_F()
264 builder2.SetOutputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT}); in TEST_F()
306 builder1.SetOutputsFormat({kOpFormat_DEFAULT}); in TEST_F()
[all …]
Dtranspose_transdata_fusion_test.cc59 builder.SetOutputsFormat({"NC1HWC0"}); in SelectKernel()
68 builder.SetOutputsFormat({"NC1HWC0"}); in SelectKernel()
104 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc68 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
83 builder1.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
139 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
162 builder1.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
175 builder2.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
229 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
252 builder1.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
265 builder2.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
311 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
323 builder2.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dtrans_op_format_refine.cc43 builder->SetOutputsFormat({out_format}); in Process()
49 builder->SetOutputsFormat({kOpFormat_NCDHW}); in Process()
Ddynamic_rnn_grad_reformat.cc52 matmul_new_builder->SetOutputsFormat({kOpFormat_FRAC_NZ}); in Process()
66 split_new_builder->SetOutputsFormat({kOpFormat_FRAC_NZ, kOpFormat_FRAC_NZ}); in Process()
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dallreduce_fusion_test.cc54 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
95 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
136 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
193 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
Deliminate_redundant_op_test.cc63 builder.SetOutputsFormat({"NC1HWC0"}); in SelectKernel()
104 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
171 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
248 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc63 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
106 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
162 …builder.SetOutputsFormat({"NC1HWC0", "NC1HWC0", "NC1HWC0", "NC1HWC0", "NC1HWC0", "NC1HWC0", "NC1HW… in TEST_F()
218 builder.SetOutputsFormat({"NC1HWC0", "NC1HWC0", "NC1HWC0"}); in TEST_F()
263 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
278 builder1.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
335 builder.SetOutputsFormat({"NC1HWC0"}); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dtransposed_update_fusion.cc59 op_builder.SetOutputsFormat({kOpFormat_DEFAULT}); in CreatePermValueNode()
99 builder->SetOutputsFormat({output_format}); in Process()
Drefresh_parameter_format.cc38 builder->SetOutputsFormat({cnode_input_format}); in DoRefresh()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc163 builder->SetOutputsFormat(output_format); in SetTensorDeviceInfo()
172 builder->SetOutputsFormat(output_format); in SetTensorDeviceInfo()
301 builder.SetOutputsFormat(outputs_format); in SetGraphKernelInfo()
333 graph_info_builder.SetOutputsFormat(graph_output_format); in SetGraphKernelInfo()
428 builder->SetOutputsFormat(outputs_format); in SetKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dtopk_split.cc76 builder1.SetOutputsFormat({kOpFormat_DEFAULT}); in CreateValueNode()
88 builder.SetOutputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT}); in CreateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_graph_kernel.cc72 builder.SetOutputsFormat(std::vector<std::string>{kOpFormat_DEFAULT}); in ResetKernelBuildInfo()
299 builder.SetOutputsFormat(outputs_format); in UpdateInputsKernelInfo()
419 builder.SetOutputsFormat(outputs_format); in UpdateFormatsAndDtypes()
440 builder.SetOutputsFormat(std::vector<std::string>{kOpFormat_DEFAULT}); in UpdateFormatsAndDtypes()
470 graph_info_builder.SetOutputsFormat(graph_output_format); in SetGraphKernelInfo()
Dkernel_select_ascend.cc356 info_builder->SetOutputsFormat({format}); in SetCastAndWeightFormat()
388 builder->SetOutputsFormat(output_format); in SetWeightFormat()
395 builder->SetOutputsFormat(output_format); in SetWeightFormat()
425 info_builder->SetOutputsFormat({format}); in RefreshCastAndParamWeightFormat()
560 builder->SetOutputsFormat(kernel_build_info->GetAllOutputFormats()); in SetKernelInfo()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/enhancer/
Dinsert_tensor_move_for_getnext_test.cc49 builder.SetOutputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT}); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_metadata.cc58 builder.SetOutputsFormat(outputs_format); in HostMetadataInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dtsa_atomic_add_to_first_tensor.cc120 para_info_builder.SetOutputsFormat({parameter_format}); in ProcessTsaFirstNode()
177 new_info_builder.SetOutputsFormat(new_outputs_format); in CorrectKernelBuildInfo()

1234