Searched refs:new_tuple_size (Results 1 – 1 of 1) sorted by relevance
143 int new_tuple_size = new_tuple->shape().tuple_shapes().size(); in FlattenWhileLoop() local149 CHECK_EQ(shape->tuple_shapes().size(), new_tuple_size - 1); in FlattenWhileLoop()197 shape, condition->parameter_instruction(0), new_tuple_size - 1)); in FlattenWhileLoop()216 shape, body->parameter_instruction(0), new_tuple_size - 1)); in FlattenWhileLoop()242 prefix = TupleUtil::ExtractPrefix(while_hlo, new_tuple_size - 1); in FlattenWhileLoop()