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