View source on GitHub |
Extracts a set of the variable names referenced by the node in question.
tfp.experimental.auto_batching.instructions.extract_referenced_variables(
node
)
Args | |
---|---|
node
|
Most structures from the VM, including ops, sequences of ops, blocks. |
Returns | |
---|---|
varnames
|
set of variable names referenced by the node in question.
|