TensorFlow 2.0 Beta is available Learn more

tff.framework.inline_block_locals

Inlines the block variables in comp whitelisted by variable_names.

tff.framework.inline_block_locals(
    comp,
    variable_names=None
)

Defined in python/core/impl/transformations.py.

Args:

  • comp: The computation building block in which to perform the extractions. The names of lambda parameters and block variables in comp must be unique.
  • variable_names: A Python list, tuple, or set representing the whitelist of variable names to inline; or None if all variables should be inlined.

Returns:

A new computation with the transformation applied or the original comp.

Raises:

  • ValueError: If comp contains variables with non-unique names.