| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils_test.cc | 33 HloComputation** entry_computation) { in CreateModuleWithProgramShape()
48 HloComputation* entry_computation; in TEST_F() local
67 HloComputation* entry_computation; in TEST_F() local
90 HloComputation* entry_computation; in TEST_F() local
109 HloComputation* entry_computation; in TEST_F() local
128 HloComputation* entry_computation; in TEST_F() local
147 HloComputation* entry_computation; in TEST_F() local
168 HloComputation* entry_computation; in TEST_F() local
188 HloComputation* entry_computation; in TEST_F() local
207 HloComputation* entry_computation; in TEST_F() local
|
| D | tuple_util_test.cc | 28 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule()
51 HloComputation* entry_computation; in TEST() local
65 HloComputation* entry_computation; in TEST() local
|
| D | while_util_test.cc | 30 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule()
66 HloComputation* entry_computation; in TEST() local
97 HloComputation* entry_computation; in TEST() 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 | 240 HloComputation* entry_computation = in TEST_F() local
276 HloComputation* entry_computation = in TEST_F() local
312 HloComputation* entry_computation = in TEST_F() local
370 HloComputation* entry_computation = in TEST_F() local
458 HloComputation* entry_computation = in TEST_F() local
527 HloComputation* entry_computation = in TEST_F() local
615 HloComputation* entry_computation = in TEST_P() local
|
| D | transpose_folding_test.cc | 176 HloComputation* entry_computation = in TEST_F() local
252 HloComputation* entry_computation = in TEST_F() local
308 HloComputation* entry_computation = in TEST_F() local
369 HloComputation* entry_computation = in TEST_F() local
436 HloComputation* entry_computation = in TEST_F() local
|
| D | hlo_dce.cc | 70 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 | hlo_module.h | 107 HloComputation* entry_computation() const { in entry_computation() function
|
| 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_memory_scheduler_test.cc | 291 auto entry_computation = module->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
|
| D | heap_simulator.cc | 79 const HloComputation* entry_computation = module.entry_computation(); in Run() local
|
| D | hlo_parser.cc | 512 HloComputation* entry_computation = nullptr; in ParseComputations() local
548 bool HloParser::ParseComputation(HloComputation** entry_computation) { in ParseComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_rewriter_test.cc | 124 HloComputation* entry_computation = in TEST_F() local
160 HloComputation* entry_computation = in TEST_F() local
190 HloComputation* entry_computation = in TEST_F() local
220 HloComputation* entry_computation = in TEST_F() local
249 HloComputation* entry_computation = in TEST_F() local
302 HloComputation* entry_computation = in TEST_F() local
350 HloComputation* entry_computation = in TEST_F() local
384 HloComputation* entry_computation = in TEST_F() local
438 HloComputation* entry_computation = in TEST_F() local
488 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 | nvptx_compiler.cc | 685 HloComputation* entry_computation = module->entry_computation(); in RunBackend() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils_test.cc | 40 HloComputation* entry_computation = module->entry_computation(); in TEST() local
|
| D | conv_canonicalization_test.cc | 91 HloComputation* entry_computation = in TEST_F() local
|
| D | cpu_compiler.cc | 496 const HloComputation& entry_computation = *module.entry_computation(); in CreateHloProfilingArtifacts() local
617 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()
|