| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils_test.cc | 33 HloComputation** entry_computation) { in CreateModuleWithProgramShape()
48 HloComputation** entry_computation, PrimitiveType primitive_type_output) { in CreateModuleWithProgramShape()
63 HloComputation* entry_computation; in TEST_F() local
82 HloComputation* entry_computation; in TEST_F() local
105 HloComputation* entry_computation; in TEST_F() local
124 HloComputation* entry_computation; in TEST_F() local
143 HloComputation* entry_computation; in TEST_F() local
162 HloComputation* entry_computation; in TEST_F() local
183 HloComputation* entry_computation; in TEST_F() local
203 HloComputation* entry_computation; in TEST_F() local
[all …]
|
| D | tuple_util_test.cc | 34 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule()
60 HloComputation* entry_computation; in TEST() local
73 HloComputation* entry_computation; in TEST() local
|
| D | while_util_test.cc | 35 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule()
71 HloComputation* entry_computation; in TEST_F() local
102 HloComputation* entry_computation; in TEST_F() local
|
| D | call_graph_test.cc | 118 HloComputation* entry_computation = in TEST_F() local
144 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
175 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
216 HloComputation* entry_computation = in TEST_F() local
266 HloComputation* entry_computation = in TEST_F() local
331 HloComputation* entry_computation; in TEST_F() local
465 HloComputation* entry_computation; in TEST_F() local
519 HloComputation* entry_computation = in TEST_F() local
|
| D | hlo_rematerialization_test.cc | 136 HloComputation* entry_computation = in TEST_F() local
172 HloComputation* entry_computation = in TEST_F() local
210 HloComputation* entry_computation = in TEST_F() local
268 HloComputation* entry_computation = in TEST_F() local
356 HloComputation* entry_computation = in TEST_F() local
425 HloComputation* entry_computation = in TEST_F() local
513 HloComputation* entry_computation = in TEST_P() local
|
| D | transpose_folding_test.cc | 175 HloComputation* entry_computation = in TEST_F() local
251 HloComputation* entry_computation = in TEST_F() local
307 HloComputation* entry_computation = in TEST_F() local
368 HloComputation* entry_computation = in TEST_F() local
435 HloComputation* entry_computation = in TEST_F() local
|
| D | hlo_dce.cc | 85 if (HloComputation* entry_computation = module->entry_computation()) { in Run() local
|
| D | hlo_profile_printer_data.proto | 65 string entry_computation = 4; field
|
| D | flatten_call_graph_test.cc | 132 HloComputation* entry_computation; in TEST_F() local
168 HloComputation* entry_computation; in TEST_F() local
|
| D | while_loop_invariant_code_motion_test.cc | 89 HloComputation* entry_computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
148 HloComputation* entry_computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
388 HloComputation* entry_computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
|
| D | hlo_module.h | 115 HloComputation* entry_computation() const { in entry_computation() function
|
| D | hlo_live_range_test.cc | 321 HloComputation* entry_computation = in TEST_F() local
|
| D | heap_simulator_test.cc | 82 HloComputation* entry_computation = in TEST_F() local
190 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
282 const string& name, std::unique_ptr<HloComputation> entry_computation, in HeapSimulatorTracker()
|
| D | memory_space_assignment_test.cc | 608 HloComputation* entry_computation = in TEST_P() local
1565 HloComputation* entry_computation = in TEST_P() local
1638 HloComputation* entry_computation = in TEST_P() local
1705 HloComputation* entry_computation = in TEST_P() local
1776 HloComputation* entry_computation = in TEST_P() local
1893 HloComputation* entry_computation = in TEST_P() local
1996 HloComputation* entry_computation = in TEST_P() local
|
| D | hlo_module.cc | 54 void HloModule::ReplaceEntryComputation(HloComputation* entry_computation) { in ReplaceEntryComputation()
|
| D | memory_space_assignment.cc | 530 HloComputation* entry_computation = module.entry_computation(); in AddInputAndOutputRequiredAssignments() local
590 const HloComputation* entry_computation = module.entry_computation(); in AreIntervalsReservedInAlternateMemory() local
|
| D | heap_simulator.cc | 98 const HloComputation* entry_computation = module.entry_computation(); in Run() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 124 HloComputation* entry_computation = in TEST_F() local
178 HloComputation* entry_computation = in TEST_F() local
240 HloComputation* entry_computation = in TEST_F() local
270 HloComputation* entry_computation = in TEST_F() local
300 HloComputation* entry_computation = in TEST_F() local
329 HloComputation* entry_computation = in TEST_F() local
382 HloComputation* entry_computation = in TEST_F() local
430 HloComputation* entry_computation = in TEST_F() local
464 HloComputation* entry_computation = in TEST_F() local
518 HloComputation* entry_computation = in TEST_F() local
[all …]
|
| D | variadic_op_splitter_test.cc | 58 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
73 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
|
| D | gpu_hlo_schedule.cc | 198 HloComputation* entry_computation = module.entry_computation(); in Build() local
|
| D | gpu_compiler.cc | 383 HloComputation* entry_computation = module->entry_computation(); in RunBackend() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils_test.cc | 44 HloComputation* entry_computation = module->entry_computation(); in TEST_F() local
|
| D | conv_canonicalization_test.cc | 91 HloComputation* entry_computation = in TEST_F() local
|
| D | cpu_compiler.cc | 481 const HloComputation& entry_computation = *module.entry_computation(); in CreateHloProfilingArtifacts() local
593 HloComputation* entry_computation = module->entry_computation(); in RunBackend() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_eigen_dot_operation_test.cc | 45 void CompileAndCheck(std::unique_ptr<HloComputation> entry_computation, in CompileAndCheck()
|