TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

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.

Aliases:

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.