Searched refs:enter_context_fn (Results 1 – 4 of 4) sorted by relevance
186 self.push(is_building_function=False, enter_context_fn=eager_mode,189 def push(self, is_building_function, enter_context_fn, device_stack): argument207 ContextSwitch(is_building_function, enter_context_fn, device_stack))
394 self.assertEqual(context_switches.stack[0].enter_context_fn,
144 self.assertIs(switch.enter_context_fn, context.eager_mode)
5578 outer_context = stack_entry.enter_context_fn