Home
last modified time | relevance | path

Searched refs:kFloat32 (Results 1 – 25 of 194) sorted by relevance

12345678

/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc52 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
69 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
70 builder.SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
84 builder1.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
112 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
113 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_bias); in TEST_F()
140 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
141 builder.SetOutputsDeviceType({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()
[all …]
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc47 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
64 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
65 builder.SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
92 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
107 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id()}); in TEST_F()
108 builder.SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
195 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, x_shp); in TEST_F()
196 auto gamma_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, gamma_shp); in TEST_F()
197 auto beta_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, beta_shp); in TEST_F()
219 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id(), kFloat32->type_id()}); in TEST_F()
[all …]
/third_party/skia/third_party/externals/tint/src/transform/
Dvertex_pulling_test.cc91 {{15, VertexStepMode::kVertex, {{VertexFormat::kFloat32, 0, 0}}}}}; in TEST_F()
160 {{4, VertexStepMode::kVertex, {{VertexFormat::kFloat32, 0, 0}}}}}; in TEST_F()
199 {{4, VertexStepMode::kInstance, {{VertexFormat::kFloat32, 0, 0}}}}}; in TEST_F()
238 {{4, VertexStepMode::kVertex, {{VertexFormat::kFloat32, 0, 0}}}}}; in TEST_F()
287 {{4, VertexStepMode::kVertex, {{VertexFormat::kFloat32, 0, 0}}}}}; in TEST_F()
339 {{VertexFormat::kFloat32, 0, 0}}, in TEST_F()
344 {{VertexFormat::kFloat32, 0, 1}}, in TEST_F()
419 {{VertexFormat::kFloat32, 0, 0}}, in TEST_F()
424 {{VertexFormat::kFloat32, 0, 1}}, in TEST_F()
496 {{VertexFormat::kFloat32, 0, 0}}, in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dallreduce_fusion_test.cc47 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
55 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
56 builder.SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
88 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
96 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
97 builder.SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
129 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
137 builder.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
138 builder.SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
180 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
[all …]
Dconvert_tuple_output_to_maketuple_test.cc46 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
47 auto h_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_h); in TEST_F()
48 auto c_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_c); in TEST_F()
49 auto w_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_w); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/common/
Drestore_abs_input_in_backed_infer_test.cc81 …auto abs_a = std::make_shared<abstract::AbstractTensor>(kFloat32, std::vector<int64_t>{2, 2, 2, 2}… in TEST_F()
82 …auto abs_c = std::make_shared<abstract::AbstractTensor>(kFloat32, std::vector<int64_t>{2, 2, 2, 2}… in TEST_F()
84 …auto abs_dynamic_a = std::make_shared<abstract::AbstractTensor>(kFloat32, std::vector<int64_t>{2, … in TEST_F()
85 …auto abs_dynamic_b = std::make_shared<abstract::AbstractTensor>(kFloat32, std::vector<int64_t>{2, … in TEST_F()
86 …auto abs_dynamic_c = std::make_shared<abstract::AbstractTensor>(kFloat32, std::vector<int64_t>{2, … in TEST_F()
87 …auto abs_dynamic_d = std::make_shared<abstract::AbstractTensor>(kFloat32, std::vector<int64_t>{2, … in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dfused_batch_norm_fusion_test.cc35 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
37 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
58 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
60 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
81 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
83 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
Dlamb_next_mv_rule_test.cc59 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
102 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
145 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
188 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
231 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
251 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
271 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
291 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
311 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
331 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
[all …]
Dadam_apply_one_fusion_test.cc52 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
90 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
128 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
166 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
204 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
224 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
244 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
264 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
284 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
304 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
Dadam_apply_one_with_decay_rule_test.cc37 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
58 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
79 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
100 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
121 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
142 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
163 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
184 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
205 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
226 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
Dlamb_next_mv_with_decay_rule_test.cc60 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
104 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
149 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
194 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
239 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
260 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
280 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
301 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
321 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
342 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
[all …]
Dbatchnorm_to_bninfer_test.cc36 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
38 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
56 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
58 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
Dbatchnormgrad_to_bninfergrad_test.cc37 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
39 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
57 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
59 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
Dmatmul_biasadd_fusion_test.cc35 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shpx); in TEST_F()
37 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shpy); in TEST_F()
39 auto bias_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_bias); in TEST_F()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc70 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
228 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
256 builder.SetOutputsDeviceType({kFloat32->type_id(), kFloat16->type_id()}); in TEST_F()
275 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat16->type_id()}); in TEST_F()
294 builder.SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
310 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
344 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
360 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
362 …args_spec_list.emplace_back(std::make_shared<abstract::AbstractTensor>(kFloat32, std::vector<int64… in TEST_F()
375 builder.SetOutputsDeviceType({kFloat32->type_id(), kFloat32->type_id(), kFloat32->type_id(), in TEST_F()
[all …]
Dkernel_graph_test.cc42 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shape); in TEST_F()
49 builder->SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
54 EXPECT_EQ(AnfAlgo::GetOutputInferDataType(new_value, 0), kFloat32->type_id()); in TEST_F()
70 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shape); in TEST_F()
78 EXPECT_EQ(AnfAlgo::GetOutputInferDataType(new_non_weight_parameter, 0), kFloat32->type_id()); in TEST_F()
80 EXPECT_EQ(AnfAlgo::GetOutputDeviceDataType(new_non_weight_parameter, 0), kFloat32->type_id()); in TEST_F()
92 EXPECT_EQ(AnfAlgo::GetOutputInferDataType(new_weight_parameter_node, 0), kFloat32->type_id()); in TEST_F()
142 auto abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shape); in TEST_F()
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Ddp_algo_test.cc180 matmul0->set_outputs_type({kFloat32}); in SetUp()
190 matmul1->set_outputs_type({kFloat32}); in SetUp()
200 matmul2->set_outputs_type({kFloat32}); in SetUp()
210 matmul3->set_outputs_type({kFloat32}); in SetUp()
220 matmul4->set_outputs_type({kFloat32}); in SetUp()
230 matmul5->set_outputs_type({kFloat32}); in SetUp()
240 matmul6->set_outputs_type({kFloat32}); in SetUp()
250 matmul7->set_outputs_type({kFloat32}); in SetUp()
260 matmul8->set_outputs_type({kFloat32}); in SetUp()
289 tmp_identity_ptr1->set_outputs_type({kFloat32}); in ConstructBatmanGraph()
[all …]
Drec_partition_test.cc74 p_tensor->tensor_type = kFloat32; in MakeTensor()
117 p_tensor_out->tensor_type = kFloat32; in MakeMatMulData()
128 p_tensor0->tensor_type = kFloat32; in MakeMatMulData()
133 p_tensor1->tensor_type = kFloat32; in MakeMatMulData()
138 p_tensor_out->tensor_type = kFloat32; in MakeMatMulData()
150 p_tensor_out->tensor_type = kFloat32; in MakeMatMulData()
161 p_tensor0->tensor_type = kFloat32; in MakeMatMulData()
166 p_tensor1->tensor_type = kFloat32; in MakeMatMulData()
171 p_tensor_out->tensor_type = kFloat32; in MakeMatMulData()
183 p_tensor_out->tensor_type = kFloat32; in MakeMatMulData()
Dedge_costmodel_test.cc69 matmul1->set_outputs_type({kFloat32}); in SetUp()
78 matmul2->set_outputs_type({kFloat32}); in SetUp()
87 matmul3->set_outputs_type({kFloat32}); in SetUp()
96 matmul4->set_outputs_type({kFloat32}); in SetUp()
105 matmul5->set_outputs_type({kFloat32}); in SetUp()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dinsert_cast_test.cc50 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
66 builder1.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
68 builder1.SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
107 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
120 builder1.SetInputsDeviceType({kFloat32->type_id()}); in TEST_F()
122 builder1.SetOutputsDeviceType({kFloat32->type_id()}); in TEST_F()
Dremove_internal_output_test.cc46 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in GetSingleOutputGraph()
55 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id()}); in GetSingleOutputGraph()
68 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in GetMutilpleOutputGraph()
86 builder.SetInputsDeviceType({kFloat32->type_id()}); in GetMutilpleOutputGraph()
105 builder.SetOutputsDeviceType({kFloat32->type_id()}); in SelectKernel()
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc59 TensorTypePtr tensor_type = std::make_shared<TensorType>(kFloat32); in CreateKernelGraph()
62 tensor::TensorPtr y_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), shp); in CreateKernelGraph()
64 tensor::TensorPtr z_tensor = std::make_shared<tensor::Tensor>(kFloat32->type_id(), shp); in CreateKernelGraph()
95 builder.SetInputsDeviceType({kFloat32->type_id(), kFloat32->type_id()}); in CreateKernelGraph()
97 builder.SetOutputsDeviceType({kFloat32->type_id()}); in CreateKernelGraph()
126 relu_builder.SetInputsDeviceType({kFloat32->type_id()}); in CreateKernelGraph()
127 relu_builder.SetOutputsDeviceType({kFloat32->type_id()}); in CreateKernelGraph()
155 auto abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shape); in CreateGraphWithExecOrder()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dsingle_batch_norm_fission_test.cc36 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
38 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
59 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
61 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
Dbatch_norm_bert_fission_test.cc35 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
37 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
75 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
77 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_y); in TEST_F()
Daddn_fission_test.cc39 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
64 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
89 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
114 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()
139 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp); in TEST_F()

12345678