| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops_test.py | 235 def replica_fn(): function
261 get_global_mpr(options.num_processes).run(replica_fn)
274 def replica_fn(): function
309 get_global_mpr(options.num_processes).run(replica_fn)
584 def replica_fn(): function
624 get_global_mpr(num_processes).run(replica_fn)
641 def replica_fn(): function
672 get_global_mpr(num_processes).run(replica_fn)
682 def replica_fn(): function
703 get_global_mpr(num_processes).run(replica_fn)
[all …]
|
| D | moving_averages_test.py | 60 def replica_fn(): function
70 var, assign = distribution.extended.call_for_each_replica(replica_fn)
87 def replica_fn(): function
96 var, assign_op = distribution.extended.call_for_each_replica(replica_fn)
170 def replica_fn(): function
180 var, assign = distribution.extended.call_for_each_replica(replica_fn)
|
| D | strategy_gather_test.py | 282 def replica_fn(per_replica_value): function
288 replica_fn = def_function.function(replica_fn)
291 strategy.run(replica_fn, args=(per_replica_value,)))
354 def replica_fn(per_replica_value): function
359 strategy.run(replica_fn, args=(next(input_iterator),)))
531 def replica_fn(value): function
539 strategy.run(def_function.function(replica_fn), args=(t0,))
541 strategy.run(replica_fn, args=(t0,))
|
| D | strategy_common_test.py | 133 def replica_fn(): function
138 return strategy.experimental_local_results(strategy.run(replica_fn))
154 def replica_fn(): function
159 return strategy.experimental_local_results(strategy.run(replica_fn))
|
| D | collective_all_reduce_strategy_test.py | 561 def replica_fn(): function
565 results = test_util.gather(strategy, strategy.run(replica_fn))
|
| D | mirrored_strategy_test.py | 1382 def replica_fn(): function
1389 return distribution.run(replica_fn)
1417 def replica_fn(): function
1420 distribution.run(replica_fn)
|
| /external/tensorflow/tensorflow/python/distribute/integration_test/ |
| D | mwms_peer_failure_test.py | 196 def replica_fn(): function
203 strategy.run(replica_fn)
207 strategy.run(replica_fn)
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape.py | 273 def replica_fn(gradient_tape, target, flattened_sources, output_gradients, function
306 replica_fn,
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | cluster_coordinator_test.py | 690 def replica_fn(input_tensor): function
695 return self.strategy.run(replica_fn, args=(next(iterator),))
711 self.strategy.run(replica_fn, args=(next(iter(input_fn())),))
718 self.strategy.run(replica_fn, args=(next(iter(input_fn())),))
933 def replica_fn(input_tensor): function
939 run_result = self.strategy.run(replica_fn, args=(input_tensor,))
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | parameter_server_training_test.py | 178 def replica_fn(iterator): function
192 self.coordinator.strategy.run(replica_fn, args=(iterator,))
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_NumPy_Keras_and_Distribution_Strategy.ipynb | 260 "def replica_fn():\n",
265 "print(strategy.run(replica_fn).values)"
|