TensorFlow 2.0 Beta is available Learn more

tff.framework.create_federated_map_or_apply

Creates a called federated map or apply depending on args placement.

tff.framework.create_federated_map_or_apply(
    fn,
    arg
)

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

      Call
     /    \

Intrinsic Tuple | [Comp, Comp]

Args:

  • fn: A computation_building_blocks.ComputationBuildingBlock to use as the function.
  • arg: A computation_building_blocks.ComputationBuildingBlock to use as the argument.

Returns:

A computation_building_blocks.Call.

Raises:

  • TypeError: If any of the types do not match.