Searched refs:_maybe_colocate_with (Results 1 – 3 of 3) sorted by relevance
111 with self._maybe_colocate_with(var):130 with self._maybe_colocate_with(var):139 def _maybe_colocate_with(self, var): member in VariableClippingOptimizer
201 def _maybe_colocate_with(self, value): member in _GraphTensorArray276 with self._maybe_colocate_with(value):349 with self._maybe_colocate_with(value):372 with self._maybe_colocate_with(value):
220 with _maybe_colocate_with(y.op, gradient_uid, colocate_gradients_with_ops):356 def _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops): # pylint: disable=invalid… function655 with _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops):