Searched defs:xla_while (Results 1 – 17 of 17) sorted by relevance
252 const HloInstruction* xla_while = in TEST_F() local321 HloInstruction* xla_while = in TEST_F() local395 HloInstruction* xla_while = module->entry_computation() in TEST_F() local
643 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, in BuildWhileInstruction_InitPointsToInterfering() local1387 auto xla_while = builder.AddInstruction( in TEST_F() local1664 auto xla_while = builder.AddInstruction( in TEST_F() local1805 auto xla_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local1828 for (const HloInstruction* xla_while : whiles) { in TEST_F() local1862 auto xla_while = builder.AddInstruction( in TEST_F() local2021 HloInstruction* xla_while = builder.AddInstruction( in BM_ParallelWhiles() local2086 HloInstruction* xla_while = builder.AddInstruction( in BM_ManyElementTuple() local
284 const HloInstruction* xla_while = use.instruction; in UsesBeforeValueDefinition() local309 const HloInstruction* xla_while = value.defining_instruction(); in UsesBeforeValueDefinition() local
392 auto xla_while = builder.AddInstruction( in TEST_F() local580 auto xla_while = builder.AddInstruction( in TEST_F() local845 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
51 const auto* xla_while = instruction; in RunWhileDCE() local
71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites()
200 auto* xla_while = callsite.instruction(); in PropagateLivenessToParameterCallers() local
178 const HloInstruction* xla_while, in IndicesToCopyForWhile()289 HloInstruction* xla_while) { in AddCopiesForWhile()
223 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile()
429 auto xla_while = builder.AddInstruction( in TEST_P() local865 auto xla_while = builder.AddInstruction( in TEST_P() local1393 auto xla_while = builder.AddInstruction( in TEST_P() local
1290 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile()2396 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
1018 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile()
992 bool HloDataflowAnalysis::UpdateWhileValueSet(HloInstruction* xla_while) { in UpdateWhileValueSet()
396 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
2515 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile()
444 auto xla_while = While(condition, body, init); in XLA_TEST_F() local