tfp.experimental.auto_batching.instructions.push_op

View source on GitHub

Returns an Op that pushes values from vars_in into vars_out.

Aliases:

tfp.experimental.auto_batching.instructions.push_op(
    vars_in,
    vars_out
)

Args:

  • 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.

Returns:

  • op: An Op that accomplishes the push.