Home
last modified time | relevance | path

Searched defs:entry_computation (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_convolution_rewriter_test.cc115 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 …]
Dhlo_schedule.cc195 const HloComputation* entry_computation = module.entry_computation(); in Build() local
Dgpu_compiler.cc539 HloComputation* entry_computation = module->entry_computation(); in RunBackend() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_util_test.cc28 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule()
51 HloComputation* entry_computation; in TEST() local
65 HloComputation* entry_computation; in TEST() local
Dwhile_util_test.cc28 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule()
64 HloComputation* entry_computation; in TEST() local
95 HloComputation* entry_computation; in TEST() local
Dtranspose_folding_test.cc75 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
Dcall_graph_test.cc116 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
Dhlo_module.h93 const HloComputation* entry_computation() const { in entry_computation() function
97 HloComputation* entry_computation() { in entry_computation() function
Dhlo_rematerialization_test.cc224 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
Dhlo_dce.cc76 if (HloComputation* entry_computation = module->entry_computation()) { in Run() local
Dhlo_scheduling_test.cc76 HloComputation* entry_computation = in TEST_F() local
Dflatten_call_graph_test.cc131 HloComputation* entry_computation; in TEST_F() local
167 HloComputation* entry_computation; in TEST_F() local
Dwhile_loop_invariant_code_motion_test.cc87 HloComputation* entry_computation = in TEST_F() local
146 HloComputation* entry_computation = in TEST_F() local
360 HloComputation* entry_computation = in TEST_F() local
Dhlo_module.cc226 const HloComputationProto* entry_computation = nullptr; in ProgramShapeFromProto() local
Dheap_simulator.cc69 const HloComputation* entry_computation = module.entry_computation(); in Run() local
Dbuffer_assignment.cc697 const HloComputation* entry_computation = in MaybeAssignBuffer() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_parallelization_preparation.cc39 HloComputation* entry_computation = module->entry_computation(); in Run() local
Dconv_canonicalization_test.cc89 HloComputation* entry_computation = in TEST_F() local
Dparallel_cpu_executable.cc412 const HloComputation& entry_computation = *module().entry_computation(); in ExecuteComputeFunctions() local
Dcpu_compiler.cc486 HloComputation* entry_computation = module->entry_computation(); in RunBackend() local
Dir_emitter.cc2554 auto* entry_computation = computation->parent()->entry_computation(); in FinishVisit() local
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
Dhlo_parser.cc282 HloComputation* entry_computation = nullptr; in ParseComputations() local
324 bool HloParser::ParseComputation(HloComputation** entry_computation) { in ParseComputation()