tfp.experimental.auto_batching.instructions.push_op

View source on GitHub

Returns an Op that pushes values from vars_in into vars_out.

vars_in Python pattern of string, the variables to read.
vars_out Python pattern of string, matching with vars_in; the variables to write to.

op An Op that accomplishes the push.