Searched refs:loop_body_generator_args (Results 1 – 1 of 1) sorted by relevance
189 std::vector<HloInstruction*> loop_body_generator_args; in MakeCountedLoopBodyComputation() local193 loop_body_generator_args.push_back(tuple_element); in MakeCountedLoopBodyComputation()196 loop_body_generator(indvar, loop_body_generator_args)); in MakeCountedLoopBodyComputation()