Searched refs:peek_objs (Results 1 – 5 of 5) sorted by relevance
113 def peek_objs(self): member in TraceableStack
85 self.assertEqual(expected_lifo_peek, list(t_stack.peek_objs()))
3641 for colocation_op in self._colocation_stack.peek_objs():4530 for device_spec in self._device_function_stack.peek_objs():5100 user_device_specs = self._device_function_stack.peek_objs()5742 for device_spec in innermost_nonempty_device_stack.peek_objs():
1070 for spec in device_stack.peek_objs())
3213 colocation_stack = tuple(default_graph._colocation_stack.peek_objs())