Missed TensorFlow World? Check out the recap. Learn more

tff.framework.check_intrinsics_whitelisted_for_reduction

View source on GitHub

Checks whitelist of intrinsics reducible to aggregate or broadcast.

tff.framework.check_intrinsics_whitelisted_for_reduction(comp)

Args:

  • comp: Instance of 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.