| /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 | transpose_folding_test.cc | 177 HloComputation* entry_computation = in TEST_F() local
254 HloComputation* entry_computation = in TEST_F() local
311 HloComputation* entry_computation = in TEST_F() local
373 HloComputation* entry_computation = in TEST_F() local
441 HloComputation* entry_computation = in TEST_F() local
|
| D | hlo_rematerialization_test.cc | 160 HloComputation* entry_computation = in TEST_F() local
196 HloComputation* entry_computation = in TEST_F() local
234 HloComputation* entry_computation = in TEST_F() local
292 HloComputation* entry_computation = in TEST_F() local
380 HloComputation* entry_computation = in TEST_F() local
449 HloComputation* entry_computation = in TEST_F() local
535 HloComputation* entry_computation = in TEST_P() local
|
| D | hlo_dce.cc | 93 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 | 122 HloComputation* entry_computation() const { in entry_computation() function
|
| D | hlo_live_range_test.cc | 320 HloComputation* entry_computation = in TEST_F() local
|
| D | heap_simulator_test.cc | 83 HloComputation* entry_computation = in TEST_F() local
191 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
286 const string& name, std::unique_ptr<HloComputation> entry_computation, in HeapSimulatorTracker()
|
| D | memory_space_assignment_test.cc | 179 const HloComputation* entry_computation = module->entry_computation(); in CheckParametersInDefaultMemory() local
798 HloComputation* entry_computation = in TEST_P() local
2687 HloComputation* entry_computation = in TEST_P() local
2760 HloComputation* entry_computation = in TEST_P() local
2827 HloComputation* entry_computation = in TEST_P() local
2899 HloComputation* entry_computation = in TEST_P() local
3016 HloComputation* entry_computation = in TEST_P() local
3119 HloComputation* entry_computation = in TEST_P() local
|
| D | hlo_module.cc | 59 void HloModule::ReplaceEntryComputation(HloComputation* entry_computation) { in ReplaceEntryComputation()
|
| D | heap_simulator.cc | 102 const HloComputation* entry_computation = module.entry_computation(); in Run() local
|
| D | memory_space_assignment.cc | 1726 HloComputation* entry_computation = module.entry_computation(); in AddInputAndOutputRequiredAssignments() local
1786 const HloComputation* entry_computation = module.entry_computation(); in AreIntervalsReservedInAlternateMemory() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter_test.cc | 125 HloComputation* entry_computation = in TEST_F() local
162 HloComputation* entry_computation = in TEST_F() local
192 HloComputation* entry_computation = in TEST_F() local
222 HloComputation* entry_computation = in TEST_F() local
251 HloComputation* entry_computation = in TEST_F() local
305 HloComputation* entry_computation = in TEST_F() local
355 HloComputation* entry_computation = in TEST_F() local
390 HloComputation* entry_computation = in TEST_F() local
445 HloComputation* entry_computation = in TEST_F() local
496 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 | 601 HloComputation* entry_computation = hlo_module->entry_computation(); in CompileModuleToLlvmIrImpl() 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 | 546 const HloComputation& entry_computation = *module.entry_computation(); in CreateHloProfilingArtifacts() local
696 HloComputation* entry_computation = module->entry_computation(); in RunBackend() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_eigen_dot_operation_test.cc | 46 void CompileAndCheck(std::unique_ptr<HloComputation> entry_computation, in CompileAndCheck()
|