Home
last modified time | relevance | path

Searched refs:CreateNewVerifiedModule (Results 1 – 25 of 83) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbroadcast_test.cc44 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
60 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
83 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
104 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
123 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
140 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
160 auto hlo_module = CreateNewVerifiedModule(); in TEST_F()
185 auto hlo_module = CreateNewVerifiedModule(); in TEST_F()
216 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
232 auto hlo_module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dcpu_gpu_fusion_test.cc89 auto hlo_module = CreateNewVerifiedModule(); in TestElementwise2D()
199 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
247 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
282 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
306 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
330 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
345 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
360 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
375 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
390 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
[all …]
H A Dcustom_call_test.cc107 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
122 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
143 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
167 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
192 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
236 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
252 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
269 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
H A Dcopy_test.cc46 auto module = CreateNewVerifiedModule(); in TestCopyOp()
99 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
120 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
146 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
178 auto module = CreateNewVerifiedModule(); in TestCopyConstantLayout021()
212 auto module = CreateNewVerifiedModule(); in TestCopyConstantLayoutR4()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc62 auto module = CreateNewVerifiedModule(); in TEST_F()
81 auto module = CreateNewVerifiedModule(); in TEST_F()
102 auto module = CreateNewVerifiedModule(); in TEST_F()
124 auto module = CreateNewVerifiedModule(); in TEST_F()
143 auto module = CreateNewVerifiedModule(); in TEST_F()
162 auto module = CreateNewVerifiedModule(); in TEST_F()
308 auto module = CreateNewVerifiedModule(); in TEST_F()
339 auto module = CreateNewVerifiedModule(); in TEST_F()
360 auto module = CreateNewVerifiedModule(); in TEST_F()
382 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dcpu_layout_assignment_test.cc75 auto module = CreateNewVerifiedModule(); in TEST_F()
116 auto module = CreateNewVerifiedModule(); in TEST_F()
160 auto module = CreateNewVerifiedModule(); in TEST_F()
194 auto module = CreateNewVerifiedModule(); in TEST_F()
234 auto module = CreateNewVerifiedModule(); in TEST_F()
366 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F()
378 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F()
390 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F()
402 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F()
414 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_test.cc81 auto module = CreateNewVerifiedModule(); in TEST_F()
90 auto module = CreateNewVerifiedModule(); in TEST_F()
106 auto module = CreateNewVerifiedModule(); in TEST_F()
139 auto module = CreateNewVerifiedModule(); in TEST_F()
282 auto module = CreateNewVerifiedModule(); in TEST_F()
302 auto module = CreateNewVerifiedModule(); in TEST_F()
324 auto module_a = CreateNewVerifiedModule(); in TEST_F()
325 auto module_b = CreateNewVerifiedModule(); in TEST_F()
527 auto module = CreateNewVerifiedModule(); in TEST_F()
538 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dhlo_computation_test.cc74 auto module = CreateNewVerifiedModule(); in TEST_F()
82 auto module = CreateNewVerifiedModule(); in TEST_F()
94 auto module = CreateNewVerifiedModule(); in TEST_F()
128 auto module = CreateNewVerifiedModule(); in TEST_F()
143 auto module = CreateNewVerifiedModule(); in TEST_F()
161 auto module = CreateNewVerifiedModule(); in TEST_F()
183 auto module = CreateNewVerifiedModule(); in TEST_F()
208 auto module = CreateNewVerifiedModule(); in TEST_F()
248 auto module = CreateNewVerifiedModule(); in TEST_F()
265 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dhlo_instruction_test.cc158 auto module = CreateNewVerifiedModule(); in TEST_F()
195 auto module = CreateNewVerifiedModule(); in TEST_F()
228 auto module = CreateNewVerifiedModule(); in TEST_F()
263 auto module = CreateNewVerifiedModule(); in TEST_F()
312 auto module = CreateNewVerifiedModule(); in TEST_F()
334 auto module = CreateNewVerifiedModule(); in TEST_F()
382 auto module = CreateNewVerifiedModule(); in TEST_F()
423 auto module = CreateNewVerifiedModule(); in TEST_F()
458 auto module = CreateNewVerifiedModule(); in TEST_F()
486 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dalgebraic_simplifier_test.cc60 auto m = CreateNewVerifiedModule(); in TEST_F()
365 auto m = CreateNewVerifiedModule(); in TEST_F()
533 auto module = CreateNewVerifiedModule(); in TEST_F()
555 auto module = CreateNewVerifiedModule(); in TEST_F()
575 auto module = CreateNewVerifiedModule(); in TEST_F()
600 auto module = CreateNewVerifiedModule(); in TEST_F()
614 auto m = CreateNewVerifiedModule(); in TEST_F()
652 auto m = CreateNewVerifiedModule(); in TEST_F()
673 auto m = CreateNewVerifiedModule(); in TEST_F()
721 auto m = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dhlo_dce_test.cc60 auto module = CreateNewVerifiedModule(); in TEST_F()
82 auto module = CreateNewVerifiedModule(); in TEST_F()
103 auto module = CreateNewVerifiedModule(); in TEST_F()
120 auto module = CreateNewVerifiedModule(); in TEST_F()
147 auto module = CreateNewVerifiedModule(); in TEST_F()
187 auto module = CreateNewVerifiedModule(); in TEST_F()
212 auto module = CreateNewVerifiedModule(); in TEST_F()
252 auto module = CreateNewVerifiedModule(); in TEST_F()
310 auto module = CreateNewVerifiedModule(); in TEST_F()
361 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dreshape_mover_test.cc41 auto m = CreateNewVerifiedModule(); in TEST_F()
79 auto m = CreateNewVerifiedModule();
110 auto m = CreateNewVerifiedModule(); in TEST_F()
137 auto m = CreateNewVerifiedModule(); in TEST_F()
185 auto m = CreateNewVerifiedModule();
230 auto m = CreateNewVerifiedModule();
267 auto m = CreateNewVerifiedModule();
320 auto m = CreateNewVerifiedModule();
360 auto m = CreateNewVerifiedModule();
389 auto m = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dtuple_simplifier_test.cc66 auto module = CreateNewVerifiedModule(); in TEST_F()
79 auto module = CreateNewVerifiedModule(); in TEST_F()
99 auto module = CreateNewVerifiedModule(); in TEST_F()
126 auto module = CreateNewVerifiedModule(); in TEST_F()
158 auto module = CreateNewVerifiedModule(); in TEST_F()
183 auto module = CreateNewVerifiedModule(); in TEST_F()
208 auto module = CreateNewVerifiedModule(); in TEST_F()
220 auto module = CreateNewVerifiedModule(); in TEST_F()
H A Dbuffer_assignment_test.cc362 auto module = CreateNewVerifiedModule(); in TEST_F()
386 auto module = CreateNewVerifiedModule(); in TEST_F()
415 auto module = CreateNewVerifiedModule(); in TEST_F()
438 auto module = CreateNewVerifiedModule(); in TEST_F()
467 auto module = CreateNewVerifiedModule(); in TEST_F()
510 auto module = CreateNewVerifiedModule(); in TEST_F()
549 auto module = CreateNewVerifiedModule(); in TEST_F()
599 auto module = CreateNewVerifiedModule(); in TEST_F()
667 auto module = CreateNewVerifiedModule(); in TEST_F()
745 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dbfloat16_normalization_test.cc112 auto module = CreateNewVerifiedModule(); in TEST_F()
140 auto module = CreateNewVerifiedModule(); in TEST_F()
170 auto module = CreateNewVerifiedModule(); in TEST_F()
197 auto module = CreateNewVerifiedModule(); in TEST_F()
239 auto module = CreateNewVerifiedModule(); in TEST_F()
278 auto module = CreateNewVerifiedModule(TestName(), /*replica_count=*/2); in TEST_F()
307 auto module = CreateNewVerifiedModule(TestName(), /*replica_count=*/2); in TEST_F()
336 auto module = CreateNewVerifiedModule(); in TEST_F()
365 auto module = CreateNewVerifiedModule(); in TEST_F()
419 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dcall_graph_test.cc98 auto module = CreateNewVerifiedModule(); in TEST_F()
118 auto module = CreateNewVerifiedModule(); in TEST_F()
142 auto module = CreateNewVerifiedModule(); in TEST_F()
175 auto module = CreateNewVerifiedModule(); in TEST_F()
210 auto module = CreateNewVerifiedModule(); in TEST_F()
254 auto module = CreateNewVerifiedModule(); in TEST_F()
316 auto module = CreateNewVerifiedModule(); in TEST_F()
446 auto module = CreateNewVerifiedModule(); in TEST_F()
507 auto module = CreateNewVerifiedModule(); in TEST_F()
522 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dhlo_cse_test.cc63 auto module = CreateNewVerifiedModule(); in TEST_F()
93 auto module = CreateNewVerifiedModule(); in TEST_F()
143 auto module = CreateNewVerifiedModule(); in TEST_F()
173 auto module = CreateNewVerifiedModule(); in TEST_F()
205 auto module = CreateNewVerifiedModule(); in TEST_F()
420 auto module = CreateNewVerifiedModule(); in TEST_F()
451 auto module = CreateNewVerifiedModule(); in TEST_F()
469 auto module = CreateNewVerifiedModule(); in TEST_F()
535 auto module = CreateNewVerifiedModule(); in TEST_F()
567 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dbfloat16_propagation_test.cc125 auto module = CreateNewVerifiedModule(); in TEST_F()
141 auto module = CreateNewVerifiedModule(); in TEST_F()
198 auto module = CreateNewVerifiedModule(); in TEST_F()
247 auto module = CreateNewVerifiedModule(); in TEST_F()
303 auto module = CreateNewVerifiedModule(); in TEST_F()
342 auto module = CreateNewVerifiedModule(); in TEST_F()
371 auto module = CreateNewVerifiedModule(); in TEST_F()
382 auto module = CreateNewVerifiedModule(); in TEST_F()
430 auto module = CreateNewVerifiedModule(); in TEST_F()
459 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Ddefuser_test.cc46 auto m = CreateNewVerifiedModule(); in TEST_F()
62 auto m = CreateNewVerifiedModule(); in TEST_F()
86 auto m = CreateNewVerifiedModule(); in TEST_F()
112 auto m = CreateNewVerifiedModule(); in TEST_F()
151 auto m = CreateNewVerifiedModule(); in TEST_F()
191 auto m = CreateNewVerifiedModule(); in TEST_F()
H A Dwhile_loop_invariant_code_motion_test.cc61 auto m = CreateNewVerifiedModule(); in TEST_F()
103 auto m = CreateNewVerifiedModule(); in TEST_F()
176 auto m = CreateNewVerifiedModule(); in TEST_F()
214 auto m = CreateNewVerifiedModule(); in TEST_F()
252 auto m = CreateNewVerifiedModule(); in TEST_F()
300 auto m = CreateNewVerifiedModule(); in TEST_F()
355 auto m = CreateNewVerifiedModule(); in TEST_F()
406 auto m = CreateNewVerifiedModule(); in TEST_F()
443 auto m = CreateNewVerifiedModule(); in TEST_F()
H A Dhlo_constant_folding_test.cc52 auto module = CreateNewVerifiedModule(); in TEST_F()
75 auto module = CreateNewVerifiedModule(); in TEST_F()
97 auto module = CreateNewVerifiedModule(); in TEST_F()
140 auto module = CreateNewVerifiedModule(); in TEST_F()
167 auto module = CreateNewVerifiedModule(); in TEST_F()
192 auto module = CreateNewVerifiedModule(); in TEST_F()
H A Dcall_inliner_test.cc53 auto module = CreateNewVerifiedModule(); in TEST_F()
81 auto module = CreateNewVerifiedModule(); in TEST_F()
122 auto module = CreateNewVerifiedModule(); in TEST_F()
147 auto module = CreateNewVerifiedModule(); in TEST_F()
186 auto module = CreateNewVerifiedModule(); in TEST_F()
H A Dconditional_simplifier_test.cc103 auto m = CreateNewVerifiedModule(); in TEST_F()
111 auto m = CreateNewVerifiedModule(); in TEST_F()
121 auto m = CreateNewVerifiedModule(); in TEST_F()
133 auto m = CreateNewVerifiedModule(); in TEST_F()
149 auto m = CreateNewVerifiedModule(); in TEST_F()
163 auto m = CreateNewVerifiedModule(); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_rewriter_test.cc125 auto module = CreateNewVerifiedModule(); in TEST_F()
162 auto module = CreateNewVerifiedModule(); in TEST_F()
192 auto module = CreateNewVerifiedModule(); in TEST_F()
222 auto module = CreateNewVerifiedModule(); in TEST_F()
251 auto module = CreateNewVerifiedModule(); in TEST_F()
305 auto module = CreateNewVerifiedModule(); in TEST_F()
355 auto module = CreateNewVerifiedModule(); in TEST_F()
390 auto module = CreateNewVerifiedModule(); in TEST_F()
445 auto module = CreateNewVerifiedModule(); in TEST_F()
496 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
H A Dinstruction_fusion_test.cc45 auto module = CreateNewVerifiedModule(); in TEST_F()
65 auto module = CreateNewVerifiedModule(); in TEST_F()
84 auto module = CreateNewVerifiedModule(); in TEST_F()
103 auto module = CreateNewVerifiedModule(); in TEST_F()
123 auto module = CreateNewVerifiedModule(); in TEST_F()
140 auto module = CreateNewVerifiedModule(); in TEST_F()
484 auto module = CreateNewVerifiedModule(); in TEST_F()
587 auto m = CreateNewVerifiedModule(); in TEST_F()
605 auto m = CreateNewVerifiedModule(); in TEST_F()
623 auto m = CreateNewVerifiedModule(); in TEST_F()

1234