Searched refs:colocate (Results 1 – 1 of 1) sorted by relevance
2498 def _testWhileGrad_ColocateGradients(self, colocate): argument2513 loop, v, colocate_gradients_with_ops=colocate)[0]2521 if not colocate and name.endswith("Square"):2524 elif colocate and "Square" in name:2536 self._testWhileGrad_ColocateGradients(colocate=False)2537 self._testWhileGrad_ColocateGradients(colocate=True)