View source on GitHub |
Returns an Op
that pushes values from vars_in
into vars_out
.
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.
|