Skip to content

Instantly share code, notes, and snippets.

@kevinrobinson
Created December 18, 2015 13:55
Show Gist options
  • Save kevinrobinson/f3f2cf8fe14753d1168d to your computer and use it in GitHub Desktop.
Save kevinrobinson/f3f2cf8fe14753d1168d to your computer and use it in GitHub Desktop.
REGISTER_OP("Assign")
.Input("ref: Ref(T)")
.Input("value: T")
.Output("output_ref: Ref(T)")
.Attr("T: type")
.Attr("validate_shape: bool = true")
.Attr("use_locking: bool = true")
.SetAllowsUninitializedInput()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment