Python DistributedVarOpの例

コード例 #1

ファイル: tpu_values.py プロジェクト: Rameezpnr/tensorflow

 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)

コード例 #2

ファイル: tpu_values.py プロジェクト: zhangshuoneu/tensorflow

 def op(self):
     return values.DistributedVarOp(self._primary.op.name,
                                    self._primary.op.graph,
                                    self._primary.op.traceback,
                                    self._primary.op.type)