Searched refs:_testWhileGrad_ColocateGradients (Results 1 – 1 of 1) sorted by relevance
2498 def _testWhileGrad_ColocateGradients(self, colocate): member in ControlFlowTest2536 self._testWhileGrad_ColocateGradients(colocate=False)2537 self._testWhileGrad_ColocateGradients(colocate=True)