Searched defs:induction_var (Results 1 – 3 of 3) sorted by relevance
220 HloInstruction* scatter, HloInstruction* induction_var, in ScatterLoopBody()412 const std::vector<HloInstruction*>& loop_state) { in ExpandInstruction()
151 const HloInstruction& gather, HloInstruction* induction_var, in GatherLoopBody()
86 Output induction_var; member198 Output induction_var; member