tfp.experimental.nn.util.make_convolution_transpose_fn_with_subkernels

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