Home
last modified time | relevance | path

Searched refs:ag_ctx (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/python/autograph/impl/
Dapi_test.py33 from tensorflow.python.autograph.core import ag_ctx
879 with ag_ctx.ControlStatusCtx(status=ag_ctx.Status.DISABLED):
884 with ag_ctx.ControlStatusCtx(status=ag_ctx.Status.ENABLED):
928 self.assertEqual(ag_ctx.control_status_ctx().status,
929 ag_ctx.Status.DISABLED)
933 self.assertEqual(ag_ctx.control_status_ctx().status,
934 ag_ctx.Status.ENABLED)
936 self.assertEqual(ag_ctx.control_status_ctx().status,
937 ag_ctx.Status.ENABLED)
939 self.assertEqual(ag_ctx.control_status_ctx().status,
[all …]
Dapi.py40 from tensorflow.python.autograph.core import ag_ctx
333 if ag_ctx.control_status_ctx().status == ag_ctx.Status.DISABLED:
472 if ag_ctx.INSPECT_SOURCE_SUPPORTED:
569 if ctx.status == ag_ctx.Status.ENABLED:
572 elif ctx.status == ag_ctx.Status.DISABLED:
574 elif ctx.status == ag_ctx.Status.UNSPECIFIED:
594 with ag_ctx.ControlStatusCtx(status=ag_ctx.Status.UNSPECIFIED):
641 with ag_ctx.ControlStatusCtx(status=ag_ctx.Status.DISABLED):
654 conversion_ctx=ag_ctx.NullCtx()):
/external/tensorflow/tensorflow/python/autograph/converters/
Dfunctions_test.py19 from tensorflow.python.autograph.core import ag_ctx
83 ag_ctx.control_status_ctx().status, ag_ctx.Status.DISABLED)
88 with ag_ctx.ControlStatusCtx(
89 ag_ctx.Status.DISABLED, converter.ConversionOptions(recursive=True)):
/external/tensorflow/tensorflow/python/autograph/core/
Dfunction_wrappers.py17 from tensorflow.python.autograph.core import ag_ctx
48 self.autograph_ctx = ag_ctx.ControlStatusCtx(ag_ctx.Status.ENABLED,
DBUILD22 "ag_ctx.py",
/external/tensorflow/tensorflow/python/ops/
Dmap_fn.py21 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx
486 ag_ctx = autograph_ctx.control_status_ctx()
487 autographed_fn = autograph.tf_convert(fn, ag_ctx)
/external/tensorflow/tensorflow/python/data/ops/
Dstructured_function.py163 ag_ctx = autograph_ctx.control_status_ctx()
171 ret = autograph.tf_convert(self._func, ag_ctx)(*nested_args)
/external/tensorflow/tensorflow/python/keras/
Dlosses.py21 from tensorflow.python.autograph.core import ag_ctx
154 call_fn = autograph.tf_convert(self.call, ag_ctx.control_status_ctx())
258 ag_fn = autograph.tf_convert(self.fn, ag_ctx.control_status_ctx())
Dmetrics.py25 from tensorflow.python.autograph.core import ag_ctx
169 control_status = ag_ctx.control_status_ctx()
184 control_status = ag_ctx.control_status_ctx()
674 ag_fn = autograph.tf_convert(self._fn, ag_ctx.control_status_ctx())
3400 ag_fn = autograph.tf_convert(self._fn, ag_ctx.control_status_ctx())
/external/tensorflow/tensorflow/python/framework/
Dops.py85 ag_ctx = LazyLoader( variable
547 if not ag_ctx.INSPECT_SOURCE_SUPPORTED:
550 elif ag_ctx.control_status_ctx().status == ag_ctx.Status.DISABLED:
553 elif ag_ctx.control_status_ctx().status == ag_ctx.Status.ENABLED:
562 if not ag_ctx.INSPECT_SOURCE_SUPPORTED:
565 elif ag_ctx.control_status_ctx().status == ag_ctx.Status.DISABLED:
568 elif ag_ctx.control_status_ctx().status == ag_ctx.Status.ENABLED:
Dops_test.py29 from tensorflow.python.autograph.core import ag_ctx
123 with ag_ctx.ControlStatusCtx(ag_ctx.Status.ENABLED):
126 with ag_ctx.ControlStatusCtx(ag_ctx.Status.DISABLED):
138 with ag_ctx.ControlStatusCtx(ag_ctx.Status.ENABLED):
142 with ag_ctx.ControlStatusCtx(ag_ctx.Status.DISABLED):
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_run.py23 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx
Dtpu_strategy.py28 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx
Ddistribute_lib.py200 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops.py22 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx
/external/tensorflow/tensorflow/python/eager/
Dfunction.py76 ag_ctx = lazy_loader.LazyLoader( variable
2682 ag_ctx.Status.ENABLED
2683 if self._autograph else ag_ctx.Status.DISABLED)
2684 with ag_ctx.ControlStatusCtx(
Dfunction_test.py32 from tensorflow.python.autograph.core import ag_ctx
3577 ag_ctx.control_status_ctx().status, ag_ctx.Status.ENABLED)
3579 prev_status = ag_ctx.control_status_ctx().status
3581 self.assertEqual(ag_ctx.control_status_ctx().status, prev_status)
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py31 from tensorflow.python.autograph.core import ag_ctx
868 call_fn = autograph.tf_convert(self.call, ag_ctx.control_status_ctx())
1190 return autograph.tf_convert(self.call, ag_ctx.control_status_ctx())
Dbase_layer_v1.py26 from tensorflow.python.autograph.core import ag_ctx
776 self.call, ag_ctx.control_status_ctx())