Searched refs:comm_fn (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_test_lib.py | 527 def _test_collective_comms(self, strategy, comm_fn, inputs, expected): argument
535 strategy.experimental_run(comm_fn, inputs))))
539 def _test_collective_comms_gradients(self, strategy, comm_fn, inputs, argument
546 y = comm_fn(x) * c
559 def _test_collective_comms_gradient_tape(self, strategy, comm_fn, inputs, argument
566 y = comm_fn(x) * c
586 comm_fn = lambda x: x + 1 function
587 run_op = strategy.experimental_run(comm_fn, inputs)
599 comm_fn = lambda x: x + 1 function
603 run_op = experimental_run(comm_fn, inputs)
[all …]
|