Home
last modified time | relevance | path

Searched refs:context_switches (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dops_test.py403 context_switches = context.context().context_switches
404 self.assertLen(context_switches.stack, 1)
405 self.assertFalse(context_switches.stack[0].is_building_function)
406 self.assertEqual(context_switches.stack[0].enter_context_fn,
Dcontext.py722 self.context_switches.push(False, eager_mode, None)
728 self.context_switches.pop()
1547 def context_switches(self): member in Context
Dcore_test.py415 self.assertEqual(len(context.context().context_switches.stack), 1)
416 switch = context.context().context_switches.stack[0]
/external/tensorflow/tensorflow/python/framework/
Dops.py5399 context.context().context_switches.push(default.building_function,
5409 context.context().context_switches.pop()
5433 for stack_entry in reversed(context.context().context_switches.stack):