Commit 7b2b67ec authored by Zachary Charles's avatar Zachary Charles Committed by tensorflow-copybara
Browse files

Add building block components, composers, and LearningProcess to the top-level learning API.

PiperOrigin-RevId: 410538549
parent 28f67e1e
......@@ -28,11 +28,16 @@ py_library(
visibility = ["//tensorflow_federated:__pkg__"],
deps = [
":client_weight_lib",
":client_works",
":composers",
":debug_measurements",
":distributors",
":federated_averaging",
":federated_evaluation",
":federated_sgd",
":finalizers",
":keras_utils",
":learning_process",
":model",
":model_update_aggregator",
":model_utils",
......
......@@ -20,15 +20,26 @@ from tensorflow_federated.python.learning import models
from tensorflow_federated.python.learning import optimizers
from tensorflow_federated.python.learning import reconstruction
from tensorflow_federated.python.learning.client_weight_lib import ClientWeighting
from tensorflow_federated.python.learning.client_works import build_model_delta_client_work
from tensorflow_federated.python.learning.client_works import ClientResult
from tensorflow_federated.python.learning.client_works import ClientWorkProcess
from tensorflow_federated.python.learning.composers import compose_learning_process
from tensorflow_federated.python.learning.composers import LearningAlgorithmState
from tensorflow_federated.python.learning.debug_measurements import add_debug_measurements
from tensorflow_federated.python.learning.distributors import build_broadcast_process
from tensorflow_federated.python.learning.distributors import DistributionProcess
from tensorflow_federated.python.learning.federated_averaging import build_federated_averaging_process
from tensorflow_federated.python.learning.federated_averaging import ClientFedAvg
from tensorflow_federated.python.learning.federated_evaluation import build_federated_evaluation
from tensorflow_federated.python.learning.federated_evaluation import build_local_evaluation
from tensorflow_federated.python.learning.federated_sgd import build_federated_sgd_process
from tensorflow_federated.python.learning.finalizers import build_apply_optimizer_finalizer
from tensorflow_federated.python.learning.finalizers import FinalizerProcess
from tensorflow_federated.python.learning.framework.optimizer_utils import state_with_new_model_weights
from tensorflow_federated.python.learning.keras_utils import federated_aggregate_keras_metric
from tensorflow_federated.python.learning.keras_utils import from_keras_model
from tensorflow_federated.python.learning.learning_process import LearningProcess
from tensorflow_federated.python.learning.learning_process import LearningProcessOutput
from tensorflow_federated.python.learning.model import BatchOutput
from tensorflow_federated.python.learning.model import Model
from tensorflow_federated.python.learning.model_update_aggregator import compression_aggregator
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment