description: Reshape op input x to be a vec of op_min_input_ndims-rank tensors.
View source on GitHub |
Reshape op
input x
to be a vec of op_min_input_ndims
-rank tensors.
tfp.experimental.nn.util.batchify_op(
op, op_min_input_ndims, x, *other_op_args
)