Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/autograph/impl/
Dapi_test.py36 from tensorflow.python.autograph.core import ag_ctx
881 with ag_ctx.ControlStatusCtx(status=ag_ctx.Status.DISABLED):
886 with ag_ctx.ControlStatusCtx(status=ag_ctx.Status.ENABLED):
930 self.assertEqual(ag_ctx.control_status_ctx().status,
931 ag_ctx.Status.DISABLED)
935 self.assertEqual(ag_ctx.control_status_ctx().status,
936 ag_ctx.Status.ENABLED)
938 self.assertEqual(ag_ctx.control_status_ctx().status,
939 ag_ctx.Status.ENABLED)
941 self.assertEqual(ag_ctx.control_status_ctx().status,
[all …]
Dapi.py46 from tensorflow.python.autograph.core import ag_ctx
357 if ag_ctx.control_status_ctx().status == ag_ctx.Status.DISABLED:
591 if ctx.status == ag_ctx.Status.ENABLED:
594 elif ctx.status == ag_ctx.Status.DISABLED:
596 elif ctx.status == ag_ctx.Status.UNSPECIFIED:
615 with ag_ctx.ControlStatusCtx(status=ag_ctx.Status.UNSPECIFIED):
663 with ag_ctx.ControlStatusCtx(status=ag_ctx.Status.DISABLED):
676 conversion_ctx=ag_ctx.NullCtx()):
/external/tensorflow/tensorflow/python/autograph/converters/
Dfunctions_test.py23 from tensorflow.python.autograph.core import ag_ctx
87 ag_ctx.control_status_ctx().status, ag_ctx.Status.DISABLED)
92 with ag_ctx.ControlStatusCtx(
93 ag_ctx.Status.DISABLED, converter.ConversionOptions(recursive=True)):
/external/tensorflow/tensorflow/python/autograph/core/
Dfunction_wrappers.py21 from tensorflow.python.autograph.core import ag_ctx
52 self.autograph_ctx = ag_ctx.ControlStatusCtx(ag_ctx.Status.ENABLED,
DBUILD22 "ag_ctx.py",
/external/tensorflow/tensorflow/python/ops/
Dmap_fn.py25 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx
489 ag_ctx = autograph_ctx.control_status_ctx()
490 autographed_fn = autograph.tf_convert(fn, ag_ctx)
/external/tensorflow/tensorflow/python/keras/
Dlosses.py25 from tensorflow.python.autograph.core import ag_ctx
157 call_fn = autograph.tf_convert(self.call, ag_ctx.control_status_ctx())
261 ag_fn = autograph.tf_convert(self.fn, ag_ctx.control_status_ctx())
Dmetrics.py30 from tensorflow.python.autograph.core import ag_ctx
177 control_status = ag_ctx.control_status_ctx()
192 control_status = ag_ctx.control_status_ctx()
642 ag_fn = autograph.tf_convert(self._fn, ag_ctx.control_status_ctx())
3271 ag_fn = autograph.tf_convert(self._fn, ag_ctx.control_status_ctx())
/external/tensorflow/tensorflow/python/framework/
Dops.py85 ag_ctx = LazyLoader( variable
495 if ag_ctx.control_status_ctx().status == ag_ctx.Status.DISABLED:
498 elif ag_ctx.control_status_ctx().status == ag_ctx.Status.ENABLED:
506 if ag_ctx.control_status_ctx().status == ag_ctx.Status.DISABLED:
508 elif ag_ctx.control_status_ctx().status == ag_ctx.Status.ENABLED:
Dops_test.py30 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.py27 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx
Dtpu_strategy.py32 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/keras/engine/
Dbase_layer.py36 from tensorflow.python.autograph.core import ag_ctx
855 call_fn = autograph.tf_convert(self.call, ag_ctx.control_status_ctx())
1159 ag_ctx.control_status_ctx())
1258 return autograph.tf_convert(self.call, ag_ctx.control_status_ctx())
Dbase_layer_v1.py31 from tensorflow.python.autograph.core import ag_ctx
776 self.call, ag_ctx.control_status_ctx())
/external/tensorflow/tensorflow/python/eager/
Dfunction.py83 ag_ctx = lazy_loader.LazyLoader( variable
3405 ag_ctx.Status.ENABLED
3406 if self._autograph else ag_ctx.Status.DISABLED)
3407 with ag_ctx.ControlStatusCtx(
Dfunction_test.py35 from tensorflow.python.autograph.core import ag_ctx
3395 ag_ctx.control_status_ctx().status, ag_ctx.Status.ENABLED)
3397 prev_status = ag_ctx.control_status_ctx().status
3399 self.assertEqual(ag_ctx.control_status_ctx().status, prev_status)
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py3488 ag_ctx = autograph_ctx.control_status_ctx()
3513 ret = autograph.tf_convert(func, ag_ctx)(*nested_args)