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

tff.framework.check_intrinsics_whitelisted_for_reduction

View source

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.