Home
last modified time | relevance | path

Searched refs:SetInputsFormat (Results 1 – 25 of 73) sorted by relevance

123

/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dtransdata_split_test.cc50 builder.SetInputsFormat({"NCHW"}); in SelectKernel()
59 builder.SetInputsFormat({"NC1HWC0"}); in SelectKernel()
78 builder.SetInputsFormat({"NCHW"}); in SelectKernel()
87 builder.SetInputsFormat({"NCHW"}); in SelectKernel()
121 builder.SetInputsFormat({"NCHW"}); in TEST_F()
169 builder.SetInputsFormat({kOpFormat_C1HWNCoC0}); in TEST_F()
Dbn_split_test.cc78 builder.SetInputsFormat( in TEST_F()
127 builder.SetInputsFormat( in TEST_F()
Dbn_grad_split_test.cc82 builder1.SetInputsFormat( in TEST_F()
148 builder1.SetInputsFormat( in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dlayer_norm_beta_gamma_backprop_fusion_test.cc47 …builder.SetInputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAUL… in Query()
77 builder1.SetInputsFormat({kOpFormat_DEFAULT}); in TEST_F()
96 …builder2.SetInputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAU… in TEST_F()
138 builder1.SetInputsFormat({kOpFormat_DEFAULT}); in TEST_F()
155 …builder2.SetInputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAU… in TEST_F()
197 builder1.SetInputsFormat({kOpFormat_DEFAULT}); in TEST_F()
214 …builder2.SetInputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAU… in TEST_F()
251 builder1.SetInputsFormat({kOpFormat_DEFAULT}); in TEST_F()
263 …builder2.SetInputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAULT, kOpFormat_DEFAU… in TEST_F()
305 builder1.SetInputsFormat({kOpFormat_DEFAULT}); in TEST_F()
[all …]
Dtranspose_transdata_fusion_test.cc57 builder.SetInputsFormat({"NCHW"}); in SelectKernel()
66 builder.SetInputsFormat({"NC1HWC0"}); in SelectKernel()
102 builder.SetInputsFormat({"NCHW"}); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dmerge_cast_to_op_test.cc51 builder.SetInputsFormat({kOpFormat_NCHW}); in Query()
58 builder.SetInputsFormat({kOpFormat_NC1HWC0}); in Query()
97 builder1.SetInputsFormat({kOpFormat_NCHW}); in TEST_F()
110 builder2.SetInputsFormat({kOpFormat_NCHW}); in TEST_F()
160 builder1.SetInputsFormat({kOpFormat_NCHW}); in TEST_F()
173 builder2.SetInputsFormat({kOpFormat_NC1HWC0}); in TEST_F()
Dcheck_consistency_test.cc67 builder1.SetInputsFormat({"NCHW"}); in TEST_F()
80 builder2.SetInputsFormat({"NC1HWC0"}); in TEST_F()
128 builder1.SetInputsFormat({"NCHW"}); in TEST_F()
141 builder2.SetInputsFormat({"NCHW"}); in TEST_F()
Dinsert_cast_test.cc57 builder.SetInputsFormat({"NC1HWC0", "NC1HWC0"}); in TEST_F()
65 builder1.SetInputsFormat({"NC1HWC0"}); in TEST_F()
114 builder.SetInputsFormat({"DefaultFormat"}); in TEST_F()
119 builder1.SetInputsFormat({"NC1HWC0"}); in TEST_F()
Dinsert_trans_op_test.cc51 builder.SetInputsFormat({format, format}); in GetSingleOutputGraph()
77 builder.SetInputsFormat({kOpFormat_DEFAULT}); in GetMutilpleOutputGraph()
97 builder.SetInputsFormat({"NCHW"}); in SelectKernel()
Dremove_internal_output_test.cc54 builder.SetInputsFormat({kOpFormat_DEFAULT, kOpFormat_DEFAULT}); in GetSingleOutputGraph()
85 builder.SetInputsFormat({kOpFormat_DEFAULT}); in GetMutilpleOutputGraph()
102 builder.SetInputsFormat({kOpFormat_NC1HWC0}); in SelectKernel()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc67 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
82 builder1.SetInputsFormat({"NC1HWC0"}); in TEST_F()
138 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
161 builder1.SetInputsFormat({"NC1HWC0"}); in TEST_F()
174 builder2.SetInputsFormat({"NC1HWC0", "NC1HWC0"}); in TEST_F()
228 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
251 builder1.SetInputsFormat({"NC1HWC0"}); in TEST_F()
264 builder2.SetInputsFormat({"NC1HWC0"}); in TEST_F()
310 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
322 builder2.SetInputsFormat({"NC1HWC0", "NC1HWC0"}); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dtrans_op_format_refine.cc42 builder->SetInputsFormat({kOpFormat_NCDHW}); in Process()
48 builder->SetInputsFormat({in_format}); in Process()
Ddynamic_rnn_grad_reformat.cc51 matmul_new_builder->SetInputsFormat({kOpFormat_FRAC_NZ, kOpFormat_FRAC_NZ}); in Process()
65 split_new_builder->SetInputsFormat({kOpFormat_FRAC_NZ}); in Process()
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dallreduce_fusion_test.cc53 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
94 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
135 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
192 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
Deliminate_redundant_op_test.cc61 builder.SetInputsFormat({"NCHW"}); in SelectKernel()
103 builder.SetInputsFormat({"NC1HWC0", "NC1HWC0"}); in TEST_F()
170 builder.SetInputsFormat({"NC1HWC0", "NC1HWC0"}); in TEST_F()
247 builder.SetInputsFormat({"NC1HWC0", "NC1HWC0"}); in TEST_F()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc62 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
105 builder.SetInputsFormat({"NC1HWC0", "NC1HWC0"}); in TEST_F()
161 builder.SetInputsFormat({"NC1HWC0", "NC1HWC0", "NC1HWC0", "NC1HWC0", "NC1HWC0", "NC1HWC0"}); in TEST_F()
217 builder.SetInputsFormat({"NC1HWC0", "NC1HWC0", "NC1HWC0"}); in TEST_F()
262 builder.SetInputsFormat({"NC1HWC0"}); in TEST_F()
277 builder1.SetInputsFormat({"NC1HWC0"}); in TEST_F()
334 builder.SetInputsFormat({"NC1HWC0", "NC1HWC0"}); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dsubstitute_dropout.cc60 uniform_real_kernel_info_builder->SetInputsFormat({kOpFormat_DEFAULT}); in PreProcess()
76 …dropout_kernel_info_builder->SetInputsFormat({old_kernel_info->GetInputFormat(0), kOpFormat_DEFAUL… in PreProcess()
Ddecrease_compute_precision.cc43 builder.SetInputsFormat({format}); in AddCastCNode()
182 graph_info_builder.SetInputsFormat(cnode_input_format); in Process()
Ddecrease_transfer_precision.cc167 graph_info_builder.SetInputsFormat(cnode_input_format); in Process_Father()
280 node_info_builder.SetInputsFormat(cnode_input_format); in Process_Son()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_metadata.cc56 builder.SetInputsFormat(inputs_format); in HostMetadataInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc72 builder.SetInputsFormat(inputs_format); in AicpuMetadataInfoForSpecialNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dtranspose_transdata_fusion.cc48 new_transdata_builder->SetInputsFormat(transpose_input_formats); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Drt_kernel_info.cc72 …kernel_build_info_builder->SetInputsFormat(std::vector<std::string>(input_num, kOpFormat_DEFAULT)); in GetRtKelInfo()
Dlabel_switch.cc85 builder.SetInputsFormat({input_format[i]}); in GetKernelInfo()
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc94 builder.SetInputsFormat({kOpFormat_NCHW, kOpFormat_NCHW}); in CreateKernelGraph()
124 relu_builder.SetInputsFormat({kOpFormat_NCHW}); in CreateKernelGraph()

123