Searched refs:distributed_variable (Results 1 – 1 of 1) sorted by relevance
1000 def __init__(self, distributed_variable, primary_variable, name): argument1001 self._distributed_variable = distributed_variable1005 tensor, spec = distributed_variable._policy.get_saveable(1006 distributed_variable, primary_variable, name)