Searched refs:distributed_variable (Results 1 – 1 of 1) sorted by relevance
453 def __init__(self, distributed_variable): argument454 self.distributed_variable = distributed_variable455 self.components = (tuple(distributed_variable.shape.as_list()),456 distributed_variable.dtype)465 return self.distributed_variable1060 def __init__(self, distributed_variable, primary_variable, name): argument1061 self._distributed_variable = distributed_variable1068 tensor, spec = distributed_variable._policy.get_saveable(1069 distributed_variable, primary_variable, name)