Searched refs:_colocation_stack (Results 1 – 4 of 4) sorted by relevance
3562 if self._colocation_stack:3564 for colocation_op in self._colocation_stack.peek_objs():4323 current_stack = self._colocation_stack4324 self._colocation_stack = traceable_stack.TraceableStack()4330 self._colocation_stack.push_obj(op, offset=4)4338 self._colocation_stack.pop_obj()4342 self._colocation_stack = current_stack4998 def _colocation_stack(self): member in Graph5006 self._thread_local._colocation_stack = stack_copy_for_this_thread5007 return self._thread_local._colocation_stack[all …]
216 self._colocation_stack = graph._colocation_stack.copy() # pylint: disable=protected-access
887 func_graph._colocation_stack = colocation_stack
1452 colocation_stack = tuple(default_graph._colocation_stack.peek_objs())