INNER CODE UNIT · Python
broadcast_global_variables
bytedance/byteps · byteps/_keras/__init__.py:93
def broadcast_global_variables(backend, root_rank):
return _eval(backend, bps.broadcast_global_variables(root_rank))
def push_pull(backend, value, name, average):
return _eval(backend, bps.push_pull(tf.constant(value, name=name), average=average))
def broadcast(backend, value, root_rank, name):
return _eval(backend, bps.broadcast(tf.constant(value, name=name), root_rank, is_variable=False))
def load_model(keras, wrap_optimizer, optimizer_modules, filepath, custom_optimizers, custom_objects):
byteps_objects = {
subclass.__name__.lower(): wrap_optimizer(subclass)
for subclass in keras.optimizers.Optimizer.__subclasses__()
if subclass.__module__ in optimizer_modules
}