ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tfp.experimental.auto_batching.instructions.extract_referenced_variables

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.