Searched defs:xla_while (Results 1 – 17 of 17) sorted by relevance
251 const HloInstruction* xla_while = in TEST_F() local320 HloInstruction* xla_while = in TEST_F() local
284 const HloInstruction* xla_while = use.instruction; in UsesBeforeValueDefinition() local309 const HloInstruction* xla_while = value.defining_instruction(); in UsesBeforeValueDefinition() local
50 const auto* xla_while = instruction; in RunWhileDCE() local
359 auto xla_while = builder.AddInstruction( in TEST_F() local548 auto xla_while = builder.AddInstruction( in TEST_F() local815 auto xla_while = builder.AddInstruction( in TEST_F() local965 auto xla_while = builder.AddInstruction( in TEST_F() local1219 auto xla_while = builder.AddInstruction( in TEST_F() local
149 auto xla_while = builder.AddInstruction( in TEST_F() local232 auto xla_while = builder.AddInstruction( in TEST_F() local
71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites()
738 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, in BuildWhileInstruction_InitPointsToInterfering() local1541 auto xla_while = builder.AddInstruction( in TEST_F() local1818 auto xla_while = builder.AddInstruction( in TEST_F() local1959 auto xla_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local1982 for (const HloInstruction* xla_while : whiles) { in TEST_F() local2016 auto xla_while = builder.AddInstruction( in TEST_F() local2175 HloInstruction* xla_while = builder.AddInstruction( in BM_ParallelWhiles() local2240 HloInstruction* xla_while = builder.AddInstruction( in BM_ManyElementTuple() local
204 auto* xla_while = callsite.instruction(); in PropagateLivenessToParameterCallers() local
225 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile()
422 auto xla_while = builder.AddInstruction( in TEST_P() local858 auto xla_while = builder.AddInstruction( in TEST_P() local1118 auto xla_while = builder.AddInstruction( in TEST_P() local1466 auto xla_while = builder.AddInstruction( in TEST_P() local
1257 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile()2167 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
174 const HloInstruction* xla_while, in IndicesToCopyForWhile()285 HloInstruction* xla_while) { in AddCopiesForWhile()
989 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile()
893 bool HloDataflowAnalysis::UpdateWhileValueSet(HloInstruction* xla_while) { in UpdateWhileValueSet()
281 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
2380 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile()
449 auto xla_while = While(condition, body, init); in XLA_TEST_F() local