Searched refs:testTrainableVariables (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | one_device_strategy_test.py | 116 def testTrainableVariables(self, distribution): member in OneDeviceStrategyTest |
| H A D | collective_all_reduce_strategy_test.py | 656 def testTrainableVariables(self, required_gpus, use_devices_arg): member in SingleWorkerCollectiveAllReduceStrategy |
| H A D | mirrored_strategy_test.py | 258 def testTrainableVariables(self, distribution): member in MirroredTwoDeviceDistributionTest |