INNER CODE UNIT · Python
__call__
horovod/horovod · horovod/_keras/__init__.py:261
def __call__(self):
return self
def _eval(backend, op_or_result):
if hvd._executing_eagerly():
return op_or_result
else:
return backend.get_session().run(op_or_result)
if hasattr(hvd, 'broadcast_global_variables'):
def broadcast_global_variables(backend, root_rank):
return _eval(backend, hvd.broadcast_global_variables(root_rank))
def allreduce(backend, value, name, average, prescale_factor, postscale_factor, op, compression):
return _eval(backend, hvd.allreduce(tf.constant(value, name=name), average=average,