Lines Matching refs:devices
132 estimator.model_fn, devices=['/gpu:0', '/gpu:1', '/gpu:2'])
136 devices=['/gpu:0', '/gpu:1', '/gpu:2'],
222 devices=['/gpu:0', '/gpu:1'])
251 self.model_fn, losses.Reduction.MEAN, devices=['/gpu:0', '/gpu:1'])
289 devices=['/gpu:0', '/gpu:1'])
312 devices=['/gpu:0', '/gpu:1'])
341 self.model_fn, losses.Reduction.MEAN, devices=['/gpu:0', '/gpu:1'])
370 self.model_fn, devices=['/gpu:0', '/gpu:1'])
385 self.model_fn, devices=['/gpu:0'])
407 self.model_fn, devices=['/gpu:0'])
435 self.model_fn, devices=['/gpu:0'])
451 self.model_fn, devices=['/gpu:0', '/gpu:1'])
467 self.model_fn, devices=['/GPU:0'])
481 self.model_fn, devices=['/gpu:0'])
532 self.model_fn, devices=['/gpu:0'])
654 devices=['/gpu:0', '/gpu:1'])
715 devices=['/gpu:0', '/gpu:1'])
812 devices=['/gpu:0', '/gpu:1'])
843 self.model_fn, devices=['/gpu:0', '/gpu:1'])
885 self.model_fn, devices=['/gpu:0', '/gpu:1'])
916 devices=['/gpu:0', '/gpu:1'],
950 devices=['/gpu:0', '/gpu:1'],
1003 devices=['/gpu:0', '/gpu:1', '/gpu:3'],
1302 devices=['/gpu:0', '/gpu:1'],
1473 self.model_fn, devices=['/gpu:0', '/gpu:1'])
