Searched refs:enter_context_fn (Results 1 – 4 of 4) sorted by relevance
212 self.push(is_building_function=False, enter_context_fn=eager_mode,215 def push(self, is_building_function, enter_context_fn, device_stack): argument233 ContextSwitch(is_building_function, enter_context_fn, device_stack))
432 self.assertEqual(context_switches.stack[0].enter_context_fn,
424 self.assertIs(switch.enter_context_fn, context.eager_mode)
5639 outer_context = stack_entry.enter_context_fn