tfp.experimental.auto_batching.instructions.extract_referenced_variables

View source on GitHub

Extracts a set of the variable names referenced by the node in question.

node Most structures from the VM, including ops, sequences of ops, blocks.

varnames set of variable names referenced by the node in question.