Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbroadcast_test.cc45 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
61 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
84 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
105 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
124 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
141 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
161 hlo_module->AddEntryComputation(builder.Build()); in TEST_F()
186 hlo_module->AddEntryComputation(builder.Build()); in TEST_F()
217 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
233 hlo_module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dcpu_gpu_fusion_test.cc122 hlo_module->AddEntryComputation(builder.Build()) in TestElementwise2D()
231 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
259 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
291 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
319 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
335 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
350 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
365 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
380 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
395 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
[all …]
H A Dcustom_call_test.cc115 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
136 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
159 module->AddEntryComputation(b.Build()); in XLA_TEST_F()
175 module->AddEntryComputation(b.Build()); in XLA_TEST_F()
205 module->AddEntryComputation(b.Build()); in XLA_TEST_F()
245 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
261 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
285 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
H A Dcopy_test.cc47 module->AddEntryComputation(std::move(computation)); in TestCopyOp()
100 module->AddEntryComputation(std::move(computation)); in XLA_TEST_F()
121 module->AddEntryComputation(std::move(computation)); in XLA_TEST_F()
147 module->AddEntryComputation(std::move(computation)); in XLA_TEST_F()
179 module->AddEntryComputation(std::move(computation)); in TestCopyConstantLayout021()
213 module->AddEntryComputation(std::move(computation)); in TestCopyConstantLayoutR4()
H A Dtoken_hlo_test.cc34 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
46 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
63 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
78 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
98 module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_inference_test.cc119 module_->AddEntryComputation(builder.Build()); in TEST_F()
140 module_->AddEntryComputation(builder.Build()); in TEST_F()
166 module_->AddEntryComputation(builder.Build()); in TEST_F()
197 module_->AddEntryComputation(builder.Build()); in TEST_F()
227 module_->AddEntryComputation(builder.Build()); in TEST_F()
259 module_->AddEntryComputation(builder.Build()); in TEST_F()
303 module_->AddEntryComputation(builder.Build()); in TEST_F()
336 module_->AddEntryComputation(builder.Build()); in TEST_F()
381 module_->AddEntryComputation(builder.Build()); in TEST_F()
418 module_->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dhlo_computation_test.cc76 module->AddEntryComputation(CreateNegateComputation()); in TEST_F()
86 module->AddEntryComputation(CreateMapComputation(negate_computation)); in TEST_F()
111 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
129 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
144 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
162 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
184 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
209 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
249 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
266 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dhlo_instruction_test.cc159 module->AddEntryComputation(builder.Build()); in TEST_F()
196 module->AddEntryComputation(builder.Build()); in TEST_F()
229 module->AddEntryComputation(builder.Build()); in TEST_F()
264 module->AddEntryComputation(builder.Build()); in TEST_F()
313 module->AddEntryComputation(builder.Build()); in TEST_F()
352 module->AddEntryComputation(builder.Build()); in TEST_F()
396 module->AddEntryComputation(builder.Build()); in TEST_F()
424 module->AddEntryComputation(builder.Build()); in TEST_F()
459 module->AddEntryComputation(builder.Build()); in TEST_F()
487 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dalgebraic_simplifier_test.cc70 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
375 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
534 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
556 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
576 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
601 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
642 m->AddEntryComputation(builder.Build()); in TEST_F()
662 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
688 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
733 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dhlo_dce_test.cc61 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
83 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
104 module->AddEntryComputation(builder.Build()); in TEST_F()
121 module->AddEntryComputation(builder.Build()); in TEST_F()
148 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
188 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
234 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
291 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
344 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
390 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dhlo_module_test.cc82 auto computation = module->AddEntryComputation(CreateConstantComputation()); in TEST_F()
91 auto computation1 = module->AddEntryComputation(CreateConstantComputation()); in TEST_F()
113 module->AddEntryComputation( in TEST_F()
159 module->AddEntryComputation(b.Build()); in TEST_F()
289 auto computation4 = module->AddEntryComputation( in TEST_F()
307 module->AddEntryComputation(builder.Build()); in TEST_F()
528 auto computation = module->AddEntryComputation(CreateConstantComputation()); in TEST_F()
539 module->AddEntryComputation(CreateConstantComputation()); in TEST_F()
557 auto computation4 = module->AddEntryComputation( in TEST_F()
580 auto computation4 = module->AddEntryComputation( in TEST_F()
[all …]
H A Dhlo_dataflow_analysis_test.cc120 module_->AddEntryComputation(builder.Build()); in TEST_P()
169 module_->AddEntryComputation(builder.Build()); in TEST_P()
225 module_->AddEntryComputation(builder.Build()); in TEST_P()
281 module_->AddEntryComputation(builder.Build()); in TEST_P()
355 module_->AddEntryComputation(builder.Build()); in TEST_P()
431 module_->AddEntryComputation(builder.Build()); in TEST_P()
537 module_->AddEntryComputation(builder.Build()); in TEST_P()
623 module_->AddEntryComputation(builder.Build()); in TEST_P()
719 module_->AddEntryComputation(builder.Build()); in TEST_P()
812 module_->AddEntryComputation(builder.Build()); in TEST_P()
[all …]
H A Dreshape_mover_test.cc55 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
98 auto computation = m->AddEntryComputation(builder.Build());
124 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
151 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
205 auto computation = m->AddEntryComputation(builder.Build());
242 auto computation = m->AddEntryComputation(builder.Build());
286 auto computation = m->AddEntryComputation(builder.Build());
332 auto computation = m->AddEntryComputation(builder.Build());
375 auto computation = m->AddEntryComputation(builder.Build());
403 auto computation = m->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dtuple_simplifier_test.cc67 module->AddEntryComputation(builder.Build()); in TEST_F()
80 module->AddEntryComputation(builder.Build()); in TEST_F()
100 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
127 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
159 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
184 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
209 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
279 entry = module->AddEntryComputation(builder.Build()); in TEST_F()
H A Dbuffer_assignment_test.cc363 module->AddEntryComputation(builder.Build()); in TEST_F()
387 module->AddEntryComputation(builder.Build()); in TEST_F()
416 module->AddEntryComputation(builder.Build()); in TEST_F()
439 module->AddEntryComputation(builder.Build()); in TEST_F()
468 module->AddEntryComputation(builder.Build()); in TEST_F()
511 module->AddEntryComputation(builder.Build()); in TEST_F()
550 module->AddEntryComputation(builder.Build()); in TEST_F()
600 module->AddEntryComputation(builder.Build()); in TEST_F()
668 module->AddEntryComputation(builder.Build()); in TEST_F()
746 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dhlo_evaluator_test.cc86 m_->AddEntryComputation(b.Build()); in TestUnaryOp()
106 m_->AddEntryComputation(b.Build()); in TestBinaryOp()
124 m_->AddEntryComputation(b.Build()); in TestTernaryOp()
209 m_->AddEntryComputation(b.Build()); in ReduceWindowMaxIotaTest()
259 m_->AddEntryComputation(b.Build()); in TEST_P()
285 m_->AddEntryComputation(b.Build()); in TEST_P()
307 m_->AddEntryComputation(b.Build()); in TEST_P()
332 m_->AddEntryComputation(b.Build()); in TEST_P()
519 m_->AddEntryComputation(b.Build()); in TEST_F()
543 m_->AddEntryComputation(b.Build()); in TEST_F()
[all …]
H A Dbfloat16_normalization_test.cc113 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
141 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
171 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
209 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
268 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
293 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
322 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
355 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
381 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
420 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dcall_graph_test.cc100 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
120 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
145 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F()
178 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F()
222 module->AddEntryComputation(builder.Build()); in TEST_F()
272 module->AddEntryComputation(builder.Build()); in TEST_F()
343 entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F()
477 entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F()
509 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
524 module->AddEntryComputation(MakeScalarComputation()); in TEST_F()
[all …]
H A Dhlo_cse_test.cc64 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
94 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
144 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
174 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
206 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
421 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
452 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
484 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
536 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
568 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dhlo_alias_analysis_test.cc159 module_->AddEntryComputation(builder.Build()); in TEST_F()
194 module_->AddEntryComputation(builder.Build()); in TEST_F()
238 module_->AddEntryComputation(builder.Build()); in TEST_F()
273 module_->AddEntryComputation(builder.Build()); in TEST_F()
316 module_->AddEntryComputation(builder.Build()); in TEST_F()
404 module_->AddEntryComputation(builder.Build()); in TEST_F()
454 module_->AddEntryComputation(builder.Build()); in TEST_F()
496 module_->AddEntryComputation(builder.Build()); in TEST_F()
582 module_->AddEntryComputation(builder.Build()); in TEST_F()
678 module_->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dbfloat16_propagation_test.cc126 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
186 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
225 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
248 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
304 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
343 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
372 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
415 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
448 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
481 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
H A Dcopy_insertion_test.cc100 module->AddEntryComputation(builder.Build()); in TEST_F()
120 module->AddEntryComputation(builder.Build()); in TEST_F()
152 module->AddEntryComputation(builder.Build()); in TEST_F()
187 module->AddEntryComputation(builder.Build()); in TEST_F()
207 module->AddEntryComputation(builder.Build()); in TEST_F()
230 module->AddEntryComputation(builder.Build()); in TEST_F()
252 module->AddEntryComputation(builder.Build()); in TEST_F()
279 module->AddEntryComputation(builder.Build()); in TEST_F()
320 module->AddEntryComputation(builder.Build()); in TEST_F()
587 module_->AddEntryComputation(builder.Build()); in BuildWhileInstruction()
[all …]
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc63 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
82 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
103 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
125 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
144 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
163 auto computation = module->AddEntryComputation(builder.Build()); in TEST_F()
309 module->AddEntryComputation(builder.Build()); in TEST_F()
340 module->AddEntryComputation(builder.Build()); in TEST_F()
361 module->AddEntryComputation(builder.Build()); in TEST_F()
383 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_rewriter_test.cc127 module->AddEntryComputation(builder.Build()); in TEST_F()
164 module->AddEntryComputation(builder.Build()); in TEST_F()
194 module->AddEntryComputation(builder.Build()); in TEST_F()
224 module->AddEntryComputation(builder.Build()); in TEST_F()
253 module->AddEntryComputation(builder.Build()); in TEST_F()
307 module->AddEntryComputation(builder.Build()); in TEST_F()
357 module->AddEntryComputation(builder.Build()); in TEST_F()
392 module->AddEntryComputation(builder.Build()); in TEST_F()
447 module->AddEntryComputation(builder.Build()); in TEST_F()
498 module->AddEntryComputation(builder.Build()); in TEST_F()
[all …]

1234