Searched refs:loop_body_generator (Results 1 – 2 of 2) sorted by relevance
179 HloInstruction*, const WhileUtil::LoopStateTy&)>& loop_body_generator) { in MakeCountedLoopBodyComputation() argument198 loop_body_generator(indvar, loop_body_generator_args)); in MakeCountedLoopBodyComputation()242 const WhileUtil::LoopBodyGeneratorTy& loop_body_generator, in MakeCountedLoop() argument255 MakeCountedLoopBodyComputation(loop_state_shape, loop_body_generator)); in MakeCountedLoop()
86 const LoopBodyGeneratorTy& loop_body_generator,