Searched refs:_colocation_stack (Results 1 – 4 of 4) sorted by relevance
3420 if self._colocation_stack:3422 for colocation_op in self._colocation_stack.peek_objs():4188 current_stack = self._colocation_stack4189 self._colocation_stack = traceable_stack.TraceableStack()4195 self._colocation_stack.push_obj(op, offset=4)4203 self._colocation_stack.pop_obj()4207 self._colocation_stack = current_stack4893 def _colocation_stack(self): member in Graph4901 self._thread_local._colocation_stack = stack_copy_for_this_thread4902 return self._thread_local._colocation_stack[all …]
240 self._colocation_stack = graph._colocation_stack.copy() # pylint: disable=protected-access
941 func_graph._colocation_stack = colocation_stack
2612 colocation_stack = tuple(default_graph._colocation_stack.peek_objs())