TensorFlow 2.0 Beta is available Learn more

tff.framework.check_intrinsics_whitelisted_for_reduction

Checks whitelist of intrinsics reducible to aggregate or broadcast.

tff.framework.check_intrinsics_whitelisted_for_reduction(comp)

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

Args:

  • comp: Instance of computation_building_blocks.ComputationBuildingBlock to check for presence of intrinsics not currently immediately reducible to FEDERATED_AGGREGATE or FEDERATED_BROADCAST, or local processing.

Raises:

  • ValueError: If we encounter an intrinsic under comp that is not whitelisted as currently reducible.