8aaabd6dfcd2f4ae5618ef08657bbba82e8bd69f,mesh_renderer/rasterize_triangles.py,,rasterize_clip_space,#,69

Before Change

  batch_size = clip_space_vertices.shape[0].value
  vertex_count = clip_space_vertices.shape[1].value

  per_image_barycentric_coordinates = []
  per_image_vertex_ids = []
  for im in range(clip_space_vertices.shape[0]):
    barycentric_coords, triangle_ids, _ = (
        rasterize_triangles_module.rasterize_triangles(

After Change

  _, per_image_barycentric_coordinates, per_image_vertex_ids = tf.while_loop(condition, iteration, [b, per_image_barycentric_coordinates, per_image_vertex_ids])

  barycentric_coordinates = tf.reshape(per_image_barycentric_coordinates.stack(), [-1, 3])
  vertex_ids = tf.reshape(per_image_vertex_ids.stack(), [-1, 3])

  // Indexes with each pixel"s clip-space triangle"s extrema (the pixel"s
  // "corner points") ids to get the relevant properties for deferred shading.
  flattened_vertex_attributes = tf.reshape(attributes,

Instances