RSVP for your your local TensorFlow Everywhere event today!

tfp.experimental.nn.util.make_convolution_fn

Like tf.nn.conv2d except applies batch of kernels to batch of x.