Searched defs:loop_body_generator (Results 1 – 1 of 1) sorted by relevance
184 HloInstruction*, const WhileUtil::LoopStateTy&)>& loop_body_generator) { in MakeCountedLoopBodyComputation()248 const WhileUtil::LoopBodyGeneratorTy& loop_body_generator, in MakeCountedLoop()290 const WhileUtil::LoopBodyGeneratorTy& loop_body_generator, in MakeCountedLoop()