Searched defs: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
539 def _test_collective_comms_gradients(self, strategy, comm_fn, inputs, argument
559 def _test_collective_comms_gradient_tape(self, strategy, comm_fn, inputs, argument
586 comm_fn = lambda x: x + 1 function
599 comm_fn = lambda x: x + 1 function
665 def _test_collective_comms(self, strategy, comm_fn, inputs, expected, argument
679 def _test_collective_comms_gradients(self, strategy, comm_fn, inputs, argument
701 def _test_collective_comms_gradient_tape(self, strategy, comm_fn, inputs, argument
|