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.frontend.instructions.extract_referenced_variables
• tfp.experimental.auto_batching.frontend.st.inst.extract_referenced_variables
• tfp.experimental.auto_batching.frontend.stack.inst.extract_referenced_variables
• tfp.experimental.auto_batching.stack_optimization.inst.extract_referenced_variables
• tfp.experimental.auto_batching.stackless.inst.extract_referenced_variables

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.