Exemplo n.º 1
0
 def op(self):
     if values_util.is_saving_non_distributed():
         return self._primary.op
     return values.DistributedVarOp(self._primary.op.name,
                                    self._primary.op.graph,
                                    self._primary.op.traceback,
                                    self._primary.op.type)
Exemplo n.º 2
0
 def op(self):
     return values.DistributedVarOp(self._primary.op.name,
                                    self._primary.op.graph,
                                    self._primary.op.traceback,
                                    self._primary.op.type)