Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tff.framework.get_map_of_unbound_references

View source

Gets a Python dict of the unbound references in comp.

tff.framework.get_map_of_unbound_references(comp)

Compuations that are equal will have the same collections of unbounded references, so it is safe to use comp as the key for this dict even though a given compuation may appear in many positions in the AST.

Args:

  • comp: The computation building block to parse.

Returns:

A Python dict of elements where keys are the compuations in comp and values are a Python set of the names of the unbound references in the subtree of that compuation.