def _check_shapes_compat(output_tensors, shape_invariants, input_tensors): for (t, shape, input_t) in zip(output_tensors, shape_invariants, input_tensors): if not control_flow_ops._ShapeLessThanOrEqual(t.shape, shape): raise ValueError( "Input tensor '%s' enters the loop with shape %s, but has " "shape %s after one iteration. To allow the shape to vary across " "iterations, use the `shape_invariants` argument of tf.while_loop to " "specify a less-specific shape." % (input_t.name, shape, t.shape))
def _check_shapes_compat(output_tensors, shape_invariants, input_tensors): for (t, shape, input_t) in zip(output_tensors, shape_invariants, input_tensors): if not control_flow_ops._ShapeLessThanOrEqual(t.shape, shape): raise ValueError( "Input tensor '%s' enters the loop with shape %s, but has " "shape %s after one iteration. To allow the shape to vary across " "iterations, use the `shape_invariants` argument of tf.while_loop to " "specify a less-specific shape." % (input_t.name, shape, t.shape))