Searched refs:_colocation_stack (Results 1 – 4 of 4) sorted by relevance
79 colocation_stack = tuple(default_graph._colocation_stack.peek_objs())
3891 if self._colocation_stack:3894 for colocation_op in self._colocation_stack.peek_objs():4676 current_stack = self._colocation_stack4677 self._colocation_stack = traceable_stack.TraceableStack()4684 self._colocation_stack.push_obj(device_only_candidate, offset=4)4685 self._colocation_stack.push_obj(op, offset=4)4695 self._colocation_stack.pop_obj()4697 self._colocation_stack.pop_obj()4701 self._colocation_stack = current_stack5398 def _colocation_stack(self): member in Graph[all …]
295 self._colocation_stack = graph._colocation_stack.copy() # pylint: disable=protected-access
989 func_graph._colocation_stack = colocation_stack