Searched defs:comm_fn (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_test_lib.py | 544 def _test_collective_comms(self, strategy, comm_fn, inputs, expected): argument
556 def _test_collective_comms_gradients(self, strategy, comm_fn, inputs, argument
576 def _test_collective_comms_gradient_tape(self, strategy, comm_fn, inputs, argument
603 comm_fn = lambda x: x + 1 function
616 comm_fn = lambda x: x + 1 function
682 def _test_collective_comms(self, strategy, comm_fn, inputs, expected, argument
696 def _test_collective_comms_gradients(self, strategy, comm_fn, inputs, argument
718 def _test_collective_comms_gradient_tape(self, strategy, comm_fn, inputs, argument
|