parameters: (op)
return_statement: return [op.inputs[1].get_shape()]
docstring: Shape function for the Assign op.
docstring_summary: Shape function for the Assign op.
docstring_tokens: Shape function for the Assign op .
function: def _AssignShape(op):
    """Shape function for the Assign op."""
    if op.get_attr("validate_shape"):
      # NOTE(mrry): Return a known shape here. This makes it awkward to
      # chain a validated-shape assignment and a reshaping assignment,
      # but that is a sufficiently niche case that supporting it does
      # not seem worthwhile.
      return [op.inputs[0].get_shape().merge_with(op.inputs[1].get_shape())]
    return [op.inputs[1].get_shape()]
function_tokens: def _AssignShape ( op ) : if op . get_attr ( "validate_shape" ) : # NOTE(mrry): Return a known shape here. This makes it awkward to # chain a validated-shape assignment and a reshaping assignment, # but that is a sufficiently niche case that supporting it does # not seem worthwhile. return [ op . inputs [ 0 ] . get_shape ( ) . merge_with ( op . inputs [ 1 ] . get_shape ( ) ) ] return [ op . inputs [ 1 ] . get_shape ( ) ]