Searched refs:peek_objs (Results 1 – 5 of 5) sorted by relevance
112 def peek_objs(self): member in TraceableStack
85 self.assertEqual(expected_lifo_peek, t_stack.peek_objs())
3564 for colocation_op in self._colocation_stack.peek_objs():4422 for device_spec in self._device_function_stack.peek_objs():4966 user_device_specs = self._device_function_stack.peek_objs()5609 for device_spec in innermost_nonempty_device_stack.peek_objs():
767 for spec in device_stack.peek_objs())
1452 colocation_stack = tuple(default_graph._colocation_stack.peek_objs())