Searched refs:enter_context_fn (Results 1 – 4 of 4) sorted by relevance
213 self.push(is_building_function=False, enter_context_fn=eager_mode,216 def push(self, is_building_function, enter_context_fn, device_stack): argument234 ContextSwitch(is_building_function, enter_context_fn, device_stack))
406 self.assertEqual(context_switches.stack[0].enter_context_fn,
419 self.assertIs(switch.enter_context_fn, context.eager_mode)
5437 outer_context = stack_entry.enter_context_fn