Searched refs:_op_to_colocate_with (Results 1 – 2 of 2) sorted by relevance
4404 op, device_only_candidate = _op_to_colocate_with(op, self)6809 def _op_to_colocate_with(v, graph): function
1937 self._op_to_colocate_with = op_to_colocate_with1940 with g.colocate_with(self._op_to_colocate_with):