Skip to content

Instantly share code, notes, and snippets.

@kevinrobinson

kevinrobinson/assign_op_def.cc

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.
Download ZIP

Revisions

  1. kevinrobinson created this gist Dec 18, 2015.
    8 changes: 8 additions & 0 deletions assign_op_def.cc
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,8 @@
    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()