Home
last modified time | relevance | path

Searched refs:embedded_computation (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc677 for (auto embedded_computation : in RunBackend() local
679 if (embedded_computation->IsFusionComputation()) { in RunBackend()
685 embedded_computation, embedded_computation->name(), in RunBackend()
687 schedule.sequence(embedded_computation).instructions()) in RunBackend()
865 for (auto embedded_computation : in CompileAheadOfTime() local
867 if (embedded_computation->IsFusionComputation()) { in CompileAheadOfTime()
873 embedded_computation, embedded_computation->name(), in CompileAheadOfTime()
875 schedule.sequence(embedded_computation).instructions()) in CompileAheadOfTime()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module.cc520 for (HloComputation* embedded_computation : in MakeComputationPostOrder()
522 if (!added_computations.contains(embedded_computation)) { in MakeComputationPostOrder()
523 post_order.push_back(embedded_computation); in MakeComputationPostOrder()
524 added_computations.insert(embedded_computation); in MakeComputationPostOrder()
Dbuffer_liveness_test.cc404 auto embedded_computation = in TEST_F() local
411 HloInstruction::CreateCall(vec_, {param}, embedded_computation)); in TEST_F()
Dhlo_dataflow_analysis_test.cc1549 auto embedded_computation = in TEST_P() local
1560 HloInstruction::CreateCall(vector_shape_, {exp}, embedded_computation)); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc125 XlaComputation CreateMapPlusN(const XlaComputation& embedded_computation, in CreateMapPlusN() argument
129 auto map = Map(&builder, {x}, embedded_computation, {}); in CreateMapPlusN()