Propagate the name on resource variable assign (#16714)

This commit is contained in:
Alexandre Passos 2018-02-05 14:02:36 -08:00 committed by Jonathan Hseu
parent a2d007b6bc
commit ce6c6c4c12

View File

@ -278,7 +278,7 @@ def assign(ref, value, validate_shape=None, use_locking=None, name=None):
return gen_state_ops.assign( return gen_state_ops.assign(
ref, value, use_locking=use_locking, name=name, ref, value, use_locking=use_locking, name=name,
validate_shape=validate_shape) validate_shape=validate_shape)
return ref.assign(value) return ref.assign(value, name=name)
@tf_export("count_up_to") @tf_export("count_up_to")