ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tfp.experimental.nn.util.im2row_index

Computes indexes into a flattened image for building im2row.