Searched refs:distributed_variables (Results 1 – 1 of 1) sorted by relevance
37 def __init__(self, distributed_variables=None, name=None, **unused_kwargs): argument48 if not distributed_variables:50 for i, var in enumerate(distributed_variables):55 self._distributed_variables = distributed_variables56 self._devices = [v.device for v in distributed_variables]60 [var.handle for var in distributed_variables])64 trainable=distributed_variables[0].trainable,65 shape=distributed_variables[0].shape,66 dtype=distributed_variables[0].dtype,68 synchronization=distributed_variables[0].synchronization,[all …]