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
47 const auto* xla_while = instruction; in RunWhileDCE() local
197 const HloInstruction* xla_while = use.instruction; in UseIsBeforeValueDefinition() local223 const HloInstruction* xla_while = value.defining_instruction(); in UseIsBeforeValueDefinition() local
365 auto xla_while = builder.AddInstruction( in TEST_F() local556 auto xla_while = builder.AddInstruction( in TEST_F() local823 auto xla_while = builder.AddInstruction( in TEST_F() local973 auto xla_while = builder.AddInstruction( in TEST_F() local1115 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
738 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, in BuildWhileInstruction_InitPointsToInterfering() local1342 auto xla_while = builder.AddInstruction( in TEST_F() local1626 auto xla_while = builder.AddInstruction( in TEST_F() local1767 auto xla_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local1790 for (const HloInstruction* xla_while : whiles) { in TEST_F() local1824 auto xla_while = builder.AddInstruction( in TEST_F() local1976 HloInstruction* xla_while = builder.AddInstruction( in BM_ParallelWhiles() local2040 HloInstruction* xla_while = builder.AddInstruction( in BM_ManyElementTuple() local
204 auto* xla_while = callsite.instruction(); in PropagateLivenessToParameterCallers() local
169 const HloInstruction* xla_while, in IndicesToCopyForWhile()257 HloInstruction* xla_while) { in AddCopiesForWhile()
195 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile()
515 auto xla_while = builder.AddInstruction( in TEST_P() local805 auto xla_while = builder.AddInstruction( in TEST_P() local1065 auto xla_while = builder.AddInstruction( in TEST_P() local1413 auto xla_while = builder.AddInstruction( in TEST_P() local
785 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile()1621 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
656 bool HloDataflowAnalysis::UpdateWhileValueSet(HloInstruction* xla_while) { in UpdateWhileValueSet()
878 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile()
214 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
2539 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile()
798 Status IrEmitterUnnested::HandleWhile(HloInstruction* xla_while) { in HandleWhile()1672 const HloInstruction* xla_while, in CheckWhileBuffersShareAllocation()
449 auto xla_while = While(condition, body, init); in XLA_TEST_F() local