Searched defs:_test_collective_comms_gradient_tape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | strategy_test_lib.py | 576 def _test_collective_comms_gradient_tape(self, strategy, comm_fn, inputs, member in OneDeviceDistributionTestBase |