Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc45 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
61 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
84 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
105 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
124 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
141 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
161 auto hlo_module = CreateNewVerifiedModule(); in TEST_F()
186 auto hlo_module = CreateNewVerifiedModule(); in TEST_F()
217 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
233 auto hlo_module = CreateNewVerifiedModule(); in TEST_F()
[all …]
Dcpu_gpu_fusion_test.cc91 auto hlo_module = CreateNewVerifiedModule(); in TestElementwise2D()
201 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
249 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
284 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
308 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
332 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
347 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
362 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
377 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
392 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F()
[all …]
Dcustom_call_test.cc82 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
97 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
118 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
142 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
167 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
211 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
Dcopy_test.cc48 auto module = CreateNewVerifiedModule(); in TestCopyOp()
101 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
122 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
148 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
180 auto module = CreateNewVerifiedModule(); in TestCopyConstantLayout021()
214 auto module = CreateNewVerifiedModule(); in TestCopyConstantLayoutR4()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_test.cc67 auto module = CreateNewVerifiedModule(); in TEST_F()
76 auto module = CreateNewVerifiedModule(); in TEST_F()
92 auto module = CreateNewVerifiedModule(); in TEST_F()
115 auto module = CreateNewVerifiedModule(); in TEST_F()
158 auto module = CreateNewVerifiedModule(); in TEST_F()
178 auto module = CreateNewVerifiedModule(); in TEST_F()
198 auto module_a = CreateNewVerifiedModule(); in TEST_F()
199 auto module_b = CreateNewVerifiedModule(); in TEST_F()
400 auto module = CreateNewVerifiedModule(); in TEST_F()
410 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
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()
160 auto module = CreateNewVerifiedModule(); in TEST_F()
179 auto module = CreateNewVerifiedModule(); in TEST_F()
201 auto module = CreateNewVerifiedModule(); in TEST_F()
226 auto module = CreateNewVerifiedModule(); in TEST_F()
266 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
Dhlo_dce_test.cc62 auto module = CreateNewVerifiedModule(); in TEST_F()
84 auto module = CreateNewVerifiedModule(); in TEST_F()
105 auto module = CreateNewVerifiedModule(); in TEST_F()
122 auto module = CreateNewVerifiedModule(); in TEST_F()
149 auto module = CreateNewVerifiedModule(); in TEST_F()
189 auto module = CreateNewVerifiedModule(); in TEST_F()
214 auto module = CreateNewVerifiedModule(); in TEST_F()
254 auto module = CreateNewVerifiedModule(); in TEST_F()
312 auto module = CreateNewVerifiedModule(); in TEST_F()
363 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
Dtuple_simplifier_test.cc68 auto module = CreateNewVerifiedModule(); in TEST_F()
81 auto module = CreateNewVerifiedModule(); in TEST_F()
101 auto module = CreateNewVerifiedModule(); in TEST_F()
128 auto module = CreateNewVerifiedModule(); in TEST_F()
160 auto module = CreateNewVerifiedModule(); in TEST_F()
185 auto module = CreateNewVerifiedModule(); in TEST_F()
210 auto module = CreateNewVerifiedModule(); in TEST_F()
222 auto module = CreateNewVerifiedModule(); in TEST_F()
Dreshape_mover_test.cc40 auto m = CreateNewVerifiedModule(); in TEST_F()
78 auto m = CreateNewVerifiedModule();
109 auto m = CreateNewVerifiedModule(); in TEST_F()
136 auto m = CreateNewVerifiedModule(); in TEST_F()
184 auto m = CreateNewVerifiedModule();
229 auto m = CreateNewVerifiedModule();
266 auto m = CreateNewVerifiedModule();
319 auto m = CreateNewVerifiedModule();
359 auto m = CreateNewVerifiedModule();
388 auto m = CreateNewVerifiedModule(); in TEST_F()
[all …]
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()
586 auto m = CreateNewVerifiedModule(); in TEST_F()
624 auto m = CreateNewVerifiedModule(); in TEST_F()
645 auto m = CreateNewVerifiedModule(); in TEST_F()
693 auto m = CreateNewVerifiedModule(); in TEST_F()
715 auto m = CreateNewVerifiedModule(); in TEST_F()
[all …]
Dhlo_instruction_test.cc157 auto module = CreateNewVerifiedModule(); in TEST_F()
194 auto module = CreateNewVerifiedModule(); in TEST_F()
227 auto module = CreateNewVerifiedModule(); in TEST_F()
262 auto module = CreateNewVerifiedModule(); in TEST_F()
311 auto module = CreateNewVerifiedModule(); in TEST_F()
333 auto module = CreateNewVerifiedModule(); in TEST_F()
381 auto module = CreateNewVerifiedModule(); in TEST_F()
422 auto module = CreateNewVerifiedModule(); in TEST_F()
457 auto module = CreateNewVerifiedModule(); in TEST_F()
485 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
Dhlo_cse_test.cc62 auto module = CreateNewVerifiedModule(); in TEST_F()
92 auto module = CreateNewVerifiedModule(); in TEST_F()
124 auto module = CreateNewVerifiedModule(); in TEST_F()
174 auto module = CreateNewVerifiedModule(); in TEST_F()
204 auto module = CreateNewVerifiedModule(); in TEST_F()
236 auto module = CreateNewVerifiedModule(); in TEST_F()
442 auto module = CreateNewVerifiedModule(); in TEST_F()
473 auto module = CreateNewVerifiedModule(); in TEST_F()
491 auto module = CreateNewVerifiedModule(); in TEST_F()
557 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
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 …]
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()
Dbuffer_assignment_test.cc347 auto module = CreateNewVerifiedModule(); in TEST_F()
371 auto module = CreateNewVerifiedModule(); in TEST_F()
400 auto module = CreateNewVerifiedModule(); in TEST_F()
423 auto module = CreateNewVerifiedModule(); in TEST_F()
452 auto module = CreateNewVerifiedModule(); in TEST_F()
495 auto module = CreateNewVerifiedModule(); in TEST_F()
534 auto module = CreateNewVerifiedModule(); in TEST_F()
584 auto module = CreateNewVerifiedModule(); in TEST_F()
650 auto module = CreateNewVerifiedModule(); in TEST_F()
728 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
Dcall_graph_test.cc97 auto module = CreateNewVerifiedModule(); in TEST_F()
117 auto module = CreateNewVerifiedModule(); in TEST_F()
141 auto module = CreateNewVerifiedModule(); in TEST_F()
172 auto module = CreateNewVerifiedModule(); in TEST_F()
205 auto module = CreateNewVerifiedModule(); in TEST_F()
249 auto module = CreateNewVerifiedModule(); in TEST_F()
311 auto module = CreateNewVerifiedModule(); in TEST_F()
442 auto module = CreateNewVerifiedModule(); in TEST_F()
503 auto module = CreateNewVerifiedModule(); in TEST_F()
518 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
Dhlo_constant_folding_test.cc52 auto module = CreateNewVerifiedModule(); in TEST_F()
74 auto module = CreateNewVerifiedModule(); in TEST_F()
96 auto module = CreateNewVerifiedModule(); in TEST_F()
139 auto module = CreateNewVerifiedModule(); in TEST_F()
166 auto module = CreateNewVerifiedModule(); in TEST_F()
191 auto module = CreateNewVerifiedModule(); in TEST_F()
Dcall_inliner_test.cc54 auto module = CreateNewVerifiedModule(); in TEST_F()
82 auto module = CreateNewVerifiedModule(); in TEST_F()
123 auto module = CreateNewVerifiedModule(); in TEST_F()
148 auto module = CreateNewVerifiedModule(); in TEST_F()
187 auto module = CreateNewVerifiedModule(); in TEST_F()
Dhlo_subcomputation_unification_test.cc68 auto module = CreateNewVerifiedModule(); in TEST_F()
95 auto module = CreateNewVerifiedModule(); in TEST_F()
125 auto module = CreateNewVerifiedModule(); in TEST_F()
156 auto module = CreateNewVerifiedModule(); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
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()
326 auto module = CreateNewVerifiedModule(); in TEST_F()
357 auto module = CreateNewVerifiedModule(); in TEST_F()
378 auto module = CreateNewVerifiedModule(); in TEST_F()
400 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
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 …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter_test.cc124 auto module = CreateNewVerifiedModule(); in TEST_F()
161 auto module = CreateNewVerifiedModule(); in TEST_F()
191 auto module = CreateNewVerifiedModule(); in TEST_F()
221 auto module = CreateNewVerifiedModule(); in TEST_F()
250 auto module = CreateNewVerifiedModule(); in TEST_F()
304 auto module = CreateNewVerifiedModule(); in TEST_F()
354 auto module = CreateNewVerifiedModule(); in TEST_F()
389 auto module = CreateNewVerifiedModule(); in TEST_F()
444 auto module = CreateNewVerifiedModule(); in TEST_F()
495 auto module = CreateNewVerifiedModule(); in TEST_F()
[all …]
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()
Dstream_assignment_test.cc33 std::unique_ptr<HloModule> CreateNewVerifiedModule() { in CreateNewVerifiedModule() function in xla::gpu::StreamAssignmentTest
58 auto module = CreateNewVerifiedModule(); in TEST_F()
79 auto module = CreateNewVerifiedModule(); in TEST_F()
123 auto module = CreateNewVerifiedModule(); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_ldg_test.cc56 auto hlo_module = CreateNewVerifiedModule(); in TEST_F()
86 auto hlo_module = CreateNewVerifiedModule(); in TEST_F()
113 auto hlo_module = CreateNewVerifiedModule(); in TEST_F()

1234