/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_convolution_rewriter_test.cc | 115 HloComputation* entry_computation = in TEST_F() local 143 HloComputation* entry_computation = in TEST_F() local 173 HloComputation* entry_computation = in TEST_F() local 203 HloComputation* entry_computation = in TEST_F() local 231 HloComputation* entry_computation = in TEST_F() local 281 HloComputation* entry_computation = in TEST_F() local 326 HloComputation* entry_computation = in TEST_F() local 358 HloComputation* entry_computation = in TEST_F() local 411 HloComputation* entry_computation = in TEST_F() local 458 HloComputation* entry_computation = in TEST_F() local [all …]
|
D | hlo_schedule.cc | 195 const HloComputation* entry_computation = module.entry_computation(); in Build() local
|
D | gpu_compiler.cc | 539 HloComputation* entry_computation = module->entry_computation(); in RunBackend() local
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
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 | 28 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule() 64 HloComputation* entry_computation; in TEST() local 95 HloComputation* entry_computation; in TEST() local
|
D | transpose_folding_test.cc | 75 HloComputation* entry_computation = in TEST_F() local 118 HloComputation* entry_computation = in TEST_F() local 153 HloComputation* entry_computation = in TEST_F() local 186 HloComputation* entry_computation = in TEST_F() local 244 HloComputation* entry_computation = in TEST_F() local 297 HloComputation* entry_computation = in TEST_F() local 355 HloComputation* entry_computation = in TEST_F() local 419 HloComputation* entry_computation = in TEST_F() local
|
D | call_graph_test.cc | 116 HloComputation* entry_computation = in TEST_F() local 140 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local 169 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local 210 HloComputation* entry_computation = in TEST_F() local 259 HloComputation* entry_computation = in TEST_F() local 321 HloComputation* entry_computation; in TEST_F() local 444 HloComputation* entry_computation; in TEST_F() local 498 HloComputation* entry_computation = in TEST_F() local
|
D | hlo_module.h | 93 const HloComputation* entry_computation() const { in entry_computation() function 97 HloComputation* entry_computation() { in entry_computation() function
|
D | hlo_rematerialization_test.cc | 224 HloComputation* entry_computation = in TEST_F() local 263 HloComputation* entry_computation = in TEST_F() local 302 HloComputation* entry_computation = in TEST_F() local 363 HloComputation* entry_computation = in TEST_F() local 453 HloComputation* entry_computation = in TEST_F() local 567 HloComputation* entry_computation = in TEST_P() local
|
D | hlo_dce.cc | 76 if (HloComputation* entry_computation = module->entry_computation()) { in Run() local
|
D | hlo_scheduling_test.cc | 76 HloComputation* entry_computation = in TEST_F() local
|
D | flatten_call_graph_test.cc | 131 HloComputation* entry_computation; in TEST_F() local 167 HloComputation* entry_computation; in TEST_F() local
|
D | while_loop_invariant_code_motion_test.cc | 87 HloComputation* entry_computation = in TEST_F() local 146 HloComputation* entry_computation = in TEST_F() local 360 HloComputation* entry_computation = in TEST_F() local
|
D | hlo_module.cc | 226 const HloComputationProto* entry_computation = nullptr; in ProgramShapeFromProto() local
|
D | heap_simulator.cc | 69 const HloComputation* entry_computation = module.entry_computation(); in Run() local
|
D | buffer_assignment.cc | 697 const HloComputation* entry_computation = in MaybeAssignBuffer() local
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_parallelization_preparation.cc | 39 HloComputation* entry_computation = module->entry_computation(); in Run() local
|
D | conv_canonicalization_test.cc | 89 HloComputation* entry_computation = in TEST_F() local
|
D | parallel_cpu_executable.cc | 412 const HloComputation& entry_computation = *module().entry_computation(); in ExecuteComputeFunctions() local
|
D | cpu_compiler.cc | 486 HloComputation* entry_computation = module->entry_computation(); in RunBackend() local
|
D | ir_emitter.cc | 2554 auto* entry_computation = computation->parent()->entry_computation(); in FinishVisit() local
|
/external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
D | hlo_parser.cc | 282 HloComputation* entry_computation = nullptr; in ParseComputations() local 324 bool HloParser::ParseComputation(HloComputation** entry_computation) { in ParseComputation()
|