Home
last modified time | relevance | path

Searched refs:AddEntryComputation (Results 1 – 25 of 82) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc46 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
62 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
85 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
106 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
125 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
142 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
162 hlo_module->AddEntryComputation(builder.Build()); in TEST_F()
187 hlo_module->AddEntryComputation(builder.Build()); in TEST_F()
218 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
234 hlo_module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dfusion_test.cc125 hlo_module->AddEntryComputation(builder.Build()) in TestElementwise2D()
234 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
262 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
294 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
322 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
338 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
353 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
368 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
383 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
398 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
[all …]
Dcustom_call_test.cc90 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
111 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
134 module->AddEntryComputation(b.Build()); in XLA_TEST_F()
150 module->AddEntryComputation(b.Build()); in XLA_TEST_F()
180 module->AddEntryComputation(b.Build()); in XLA_TEST_F()
Dcopy_test.cc49 module->AddEntryComputation(std::move(computation)); in TestCopyOp()
102 module->AddEntryComputation(std::move(computation)); in XLA_TEST_F()
123 module->AddEntryComputation(std::move(computation)); in XLA_TEST_F()
149 module->AddEntryComputation(std::move(computation)); in XLA_TEST_F()
181 module->AddEntryComputation(std::move(computation)); in TestCopyConstantLayout021()
215 module->AddEntryComputation(std::move(computation)); in TestCopyConstantLayoutR4()
Dtoken_hlo_test.cc36 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
48 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
65 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
80 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
100 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_computation_test.cc77 module->AddEntryComputation(CreateNegateComputation()); in TEST_F()
87 module->AddEntryComputation(CreateMapComputation(negate_computation)); in TEST_F()
112 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
130 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
145 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
162 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
181 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
203 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
228 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
268 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dalgebraic_simplifier_test.cc67 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
290 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
312 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
334 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
354 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
392 m->AddEntryComputation(builder.Build()); in TEST_F()
412 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
438 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
462 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
498 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dreduce_precision_insertion_test.cc58 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
85 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
115 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
144 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
177 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
209 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
246 module->AddEntryComputation(builder.Build()); in TEST_F()
273 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
299 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
325 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Ddynamic_dimension_inference_test.cc89 module_->AddEntryComputation(builder.Build()); in TEST_F()
110 module_->AddEntryComputation(builder.Build()); in TEST_F()
136 module_->AddEntryComputation(builder.Build()); in TEST_F()
167 module_->AddEntryComputation(builder.Build()); in TEST_F()
197 module_->AddEntryComputation(builder.Build()); in TEST_F()
229 module_->AddEntryComputation(builder.Build()); in TEST_F()
265 module_->AddEntryComputation(builder.Build()); in TEST_F()
317 module_->AddEntryComputation(builder.Build()); in TEST_F()
353 module_->AddEntryComputation(builder.Build()); in TEST_F()
388 module_->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dtuple_simplifier_test.cc69 module->AddEntryComputation(builder.Build()); in TEST_F()
82 module->AddEntryComputation(builder.Build()); in TEST_F()
102 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
129 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
161 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
186 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
211 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
281 entry = module->AddEntryComputation(builder.Build()); in TEST_F()
Dhlo_instruction_test.cc156 module->AddEntryComputation(builder.Build()); in TEST_F()
193 module->AddEntryComputation(builder.Build()); in TEST_F()
226 module->AddEntryComputation(builder.Build()); in TEST_F()
261 module->AddEntryComputation(builder.Build()); in TEST_F()
310 module->AddEntryComputation(builder.Build()); in TEST_F()
349 module->AddEntryComputation(builder.Build()); in TEST_F()
393 module->AddEntryComputation(builder.Build()); in TEST_F()
421 module->AddEntryComputation(builder.Build()); in TEST_F()
456 module->AddEntryComputation(builder.Build()); in TEST_F()
484 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dreshape_mover_test.cc54 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
97 auto computation = m->AddEntryComputation(builder.Build());
123 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
150 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
204 auto computation = m->AddEntryComputation(builder.Build());
241 auto computation = m->AddEntryComputation(builder.Build());
285 auto computation = m->AddEntryComputation(builder.Build());
331 auto computation = m->AddEntryComputation(builder.Build());
374 auto computation = m->AddEntryComputation(builder.Build());
402 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dhlo_dce_test.cc61 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
82 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
112 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
152 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
198 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
253 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
308 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
354 module->AddEntryComputation(builder.Build()); in TEST_F()
401 module->AddEntryComputation(builder.Build()); in TEST_F()
Dhlo_evaluator_test.cc87 m_->AddEntryComputation(b.Build()); in TestUnaryOp()
107 m_->AddEntryComputation(b.Build()); in TestBinaryOp()
125 m_->AddEntryComputation(b.Build()); in TestTernaryOp()
164 m_->AddEntryComputation(b.Build()); in TEST_P()
190 m_->AddEntryComputation(b.Build()); in TEST_P()
212 m_->AddEntryComputation(b.Build()); in TEST_P()
237 m_->AddEntryComputation(b.Build()); in TEST_P()
405 m_->AddEntryComputation(b.Build()); in TEST_F()
429 m_->AddEntryComputation(b.Build()); in TEST_F()
450 m_->AddEntryComputation(b.Build()); in TEST_F()
[all …]
Dhlo_cse_test.cc63 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
93 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
125 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
175 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
205 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
237 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
443 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
474 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
506 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
558 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dbuffer_assignment_test.cc339 module->AddEntryComputation(builder.Build()); in TEST_F()
363 module->AddEntryComputation(builder.Build()); in TEST_F()
392 module->AddEntryComputation(builder.Build()); in TEST_F()
415 module->AddEntryComputation(builder.Build()); in TEST_F()
444 module->AddEntryComputation(builder.Build()); in TEST_F()
487 module->AddEntryComputation(builder.Build()); in TEST_F()
527 module->AddEntryComputation(builder.Build()); in TEST_F()
577 module->AddEntryComputation(builder.Build()); in TEST_F()
638 module->AddEntryComputation(builder.Build()); in TEST_F()
710 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Ddefuser_test.cc55 m->AddEntryComputation(builder.Build()); in TEST_F()
71 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
97 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
135 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
174 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
202 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
Dhlo_module_test.cc67 auto computation = module->AddEntryComputation(CreateConstantComputation()); in TEST_F()
76 auto computation1 = module->AddEntryComputation(CreateConstantComputation()); in TEST_F()
98 module->AddEntryComputation( in TEST_F()
134 module->AddEntryComputation(b.Build()); in TEST_F()
164 auto computation4 = module->AddEntryComputation( in TEST_F()
182 module->AddEntryComputation(builder.Build()); in TEST_F()
Dbuffer_liveness_test.cc121 module->AddEntryComputation(builder.Build()); in TEST_F()
168 HloComputation* entry = module->AddEntryComputation(builder.Build()); in TEST_F()
217 module->AddEntryComputation(builder.Build()); in TEST_F()
251 module->AddEntryComputation(builder.Build()); in TEST_F()
293 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
340 auto computation = module->AddEntryComputation(builder.Build(add)); in TEST_F()
377 module->AddEntryComputation(builder.Build()); in TEST_F()
413 module->AddEntryComputation(builder.Build()); in TEST_F()
454 module->AddEntryComputation(builder.Build()); in TEST_F()
518 module->AddEntryComputation(BuildDummyComputation()); in TEST_F()
[all …]
Dcopy_insertion_test.cc98 module->AddEntryComputation(builder.Build()); in TEST_F()
118 module->AddEntryComputation(builder.Build()); in TEST_F()
150 module->AddEntryComputation(builder.Build()); in TEST_F()
185 module->AddEntryComputation(builder.Build()); in TEST_F()
221 module->AddEntryComputation(builder.Build()); in TEST_F()
242 module->AddEntryComputation(builder.Build()); in TEST_F()
265 module->AddEntryComputation(builder.Build()); in TEST_F()
287 module->AddEntryComputation(builder.Build()); in TEST_F()
314 module->AddEntryComputation(builder.Build()); in TEST_F()
355 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dcall_graph_test.cc99 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
119 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
144 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F()
175 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F()
217 module->AddEntryComputation(builder.Build()); in TEST_F()
267 module->AddEntryComputation(builder.Build()); in TEST_F()
338 entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F()
473 entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F()
505 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
520 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
[all …]
Dhlo_constant_folding_test.cc51 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
73 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
95 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
138 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
165 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
190 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
Dhlo_rematerialization_test.cc167 module->AddEntryComputation(MakeRematerializableComputation()); in TEST_F()
208 module->AddEntryComputation(MakeRematerializableComputation()); in TEST_F()
241 module->AddEntryComputation(MakeRematerializableWhileComputation( in TEST_F()
277 module->AddEntryComputation(MakeRematerializableWhileComputation( in TEST_F()
313 module->AddEntryComputation(MakeRematerializableWhileComputation( in TEST_F()
371 module->AddEntryComputation(builder.Build()); in TEST_F()
459 module->AddEntryComputation(builder.Build()); in TEST_F()
528 module->AddEntryComputation(builder.Build()); in TEST_F()
616 module->AddEntryComputation(builder.Build()); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc62 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
81 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
102 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
123 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
142 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
161 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
325 module->AddEntryComputation(builder.Build()); in TEST_F()
356 module->AddEntryComputation(builder.Build()); in TEST_F()
377 module->AddEntryComputation(builder.Build()); in TEST_F()
399 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter_test.cc125 module->AddEntryComputation(builder.Build()); in TEST_F()
161 module->AddEntryComputation(builder.Build()); in TEST_F()
191 module->AddEntryComputation(builder.Build()); in TEST_F()
221 module->AddEntryComputation(builder.Build()); in TEST_F()
250 module->AddEntryComputation(builder.Build()); in TEST_F()
303 module->AddEntryComputation(builder.Build()); in TEST_F()
351 module->AddEntryComputation(builder.Build()); in TEST_F()
385 module->AddEntryComputation(builder.Build()); in TEST_F()
439 module->AddEntryComputation(builder.Build()); in TEST_F()
489 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]

1234