Save the date! Google I/O returns May 18-20 Register now

tfp.experimental.nn.util.im2row_index

Computes indexes into a flattened image for building im2row.