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.

View aliases

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