Home
last modified time | relevance | path

Searched defs:xla_while (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_schedule_test.cc252 const HloInstruction* xla_while = in TEST_F() local
321 HloInstruction* xla_while = in TEST_F() local
Dhlo_module_dce.cc47 const auto* xla_while = instruction; in RunWhileDCE() local
Dhlo_ordering.cc195 const HloInstruction* xla_while = use.instruction; in UseIsBeforeValueDefinition() local
210 const HloInstruction* xla_while = value.defining_instruction(); in UseIsBeforeValueDefinition() local
Dhlo_alias_analysis_test.cc365 auto xla_while = builder.AddInstruction( in TEST_F() local
556 auto xla_while = builder.AddInstruction( in TEST_F() local
823 auto xla_while = builder.AddInstruction( in TEST_F() local
973 auto xla_while = builder.AddInstruction( in TEST_F() local
1067 auto xla_while = builder.AddInstruction( in TEST_F() local
Dhlo_ordering_test.cc150 auto xla_while = builder.AddInstruction( in TEST_F() local
233 auto xla_while = builder.AddInstruction( in TEST_F() local
Dcopy_insertion_test.cc732 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, in BuildWhileInstruction_InitPointsToInterfering() local
1331 auto xla_while = builder.AddInstruction( in TEST_F() local
1615 auto xla_while = builder.AddInstruction( in TEST_F() local
1756 auto xla_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
1779 for (const HloInstruction* xla_while : whiles) { in TEST_F() local
1813 auto xla_while = builder.AddInstruction( in TEST_F() local
1965 HloInstruction* xla_while = builder.AddInstruction( in BM_ParallelWhiles() local
2029 HloInstruction* xla_while = builder.AddInstruction( in BM_ManyElementTuple() local
Dhlo_liveness_analysis.cc204 auto* xla_while = callsite.instruction(); in PropagateLivenessToParameterCallers() local
Ddfs_hlo_visitor_with_default.h188 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile()
Dcopy_insertion.cc169 const HloInstruction* xla_while, in IndicesToCopyForWhile()
257 HloInstruction* xla_while) { in AddCopiesForWhile()
Dhlo_verifier.cc646 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile()
1353 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
Dhlo_dataflow_analysis_test.cc515 auto xla_while = builder.AddInstruction( in TEST_P() local
805 auto xla_while = builder.AddInstruction( in TEST_P() local
1065 auto xla_while = builder.AddInstruction( in TEST_P() local
1358 auto xla_while = builder.AddInstruction( in TEST_P() local
Dhlo_cost_analysis.cc663 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile()
Dhlo_dataflow_analysis.cc586 bool HloDataflowAnalysis::UpdateWhileValueSet(HloInstruction* xla_while) { in UpdateWhileValueSet()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc88 auto xla_while = builder.AddInstruction( in TEST_F() local
Dcpu_compiler.cc226 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
Dir_emitter.cc2296 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dwhile_test.cc449 auto xla_while = While(condition, body, init); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc977 Status IrEmitterUnnested::HandleWhile(HloInstruction* xla_while) { in HandleWhile()
2055 const HloInstruction* xla_while, in CheckWhileBuffersShareAllocation()