Searched refs:_colocate_with_for_gradient (Results 1 – 2 of 2) sorted by relevance
296 with ops._colocate_with_for_gradient(op, gradient_uid): # pylint: disable=protected-access680 with ops._colocate_with_for_gradient( # pylint: disable=protected-access861 with ops._colocate_with_for_gradient( # pylint: disable=protected-access
4124 def _colocate_with_for_gradient(self, op, gradient_uid, member in Graph5116 def _colocate_with_for_gradient(op, gradient_uid, ignore_existing=False): function5132 return default_graph._colocate_with_for_gradient(5140 return _colocate_with_for_gradient(op, None, ignore_existing=ignore_existing)