Searched refs:distributed_variables (Results 1 – 1 of 1) sorted by relevance
33 def __init__(self, distributed_variables=None, name=None, **unused_kwargs): argument44 if not distributed_variables:46 for i, var in enumerate(distributed_variables):51 self._distributed_variables = distributed_variables52 self._devices = [v.device for v in distributed_variables]56 [var.handle for var in distributed_variables])60 trainable=distributed_variables[0].trainable,61 shape=distributed_variables[0].shape,62 dtype=distributed_variables[0].dtype,64 synchronization=distributed_variables[0].synchronization,[all …]