Searched refs:_colocation_stack (Results 1 – 4 of 4) sorted by relevance
3638 if self._colocation_stack:3641 for colocation_op in self._colocation_stack.peek_objs():4417 current_stack = self._colocation_stack4418 self._colocation_stack = traceable_stack.TraceableStack()4425 self._colocation_stack.push_obj(device_only_candidate, offset=4)4426 self._colocation_stack.push_obj(op, offset=4)4436 self._colocation_stack.pop_obj()4438 self._colocation_stack.pop_obj()4442 self._colocation_stack = current_stack5131 def _colocation_stack(self): member in Graph[all …]
245 self._colocation_stack = graph._colocation_stack.copy() # pylint: disable=protected-access
955 func_graph._colocation_stack = colocation_stack
3213 colocation_stack = tuple(default_graph._colocation_stack.peek_objs())