Some implementations of set_shape, when the shape of the tensors is known at tracing time, for correct compilation with tf.function.
A new implementation of utils_tf.repeat, that does not require creating variable shape tensors, when the shape of the tensors is known at tracing time.
Includes:
tf.function
.utils_tf.repeat
, that does not require creating variable shape tensors, when the shape of the tensors is known at tracing time.