| D | central_storage_strategy.py | 7 # http://www.apache.org/licenses/LICENSE-2.0 25 """A one-machine strategy that puts all variables on a single device. 28 than one GPU, compute operations (other than variable update operations) 47 strategy.run(train_step, args=(x,)) 73 …mental_distribute_dataset(self, dataset, options=None): # pylint: disable=useless-super-delegation 111 def experimental_local_results(self, value): # pylint: disable=useless-super-delegation 112 """Returns the list of all local per-replica values contained in `value`. 118 value: A value returned by `run()`, `extended.call_for_each_replica()`, 127 def run(self, fn, args=(), kwargs=None, options=None): # pylint: disable=useless-super-delegation member in CentralStorageStrategy 128 """Run `fn` on each replica, with the given arguments. [all …]
|