Searched refs:autograph_ctx (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/python/autograph/core/ |
D | function_wrappers.py | 52 self.autograph_ctx = ag_ctx.ControlStatusCtx(ag_ctx.Status.ENABLED, 76 self.autograph_ctx.__enter__() 85 self.autograph_ctx.__exit__(exc_type, exc_val, exc_tb)
|
/external/tensorflow/tensorflow/python/ops/ |
D | map_fn.py | 25 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx unknown 489 ag_ctx = autograph_ctx.control_status_ctx()
|
/external/tensorflow/tensorflow/python/distribute/ |
D | mirrored_run.py | 27 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx unknown 91 fn = autograph.tf_convert(fn, autograph_ctx.control_status_ctx())
|
D | distribute_lib.py | 200 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx unknown 1265 fn, autograph_ctx.control_status_ctx(), convert_by_default=False) 2536 fn, autograph_ctx.control_status_ctx(), convert_by_default=False) 2817 fn, autograph_ctx.control_status_ctx(), convert_by_default=False) 2984 merge_fn, autograph_ctx.control_status_ctx(), convert_by_default=False)
|
D | tpu_strategy.py | 32 from tensorflow.python.autograph.core import ag_ctx as autograph_ctx unknown 400 fn = autograph.tf_convert(fn, autograph_ctx.control_status_ctx()) 663 fn = autograph.tf_convert(fn, autograph_ctx.control_status_ctx()) 777 fn = autograph.tf_convert(fn, autograph_ctx.control_status_ctx())
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | dataset_ops.py | 87 autograph_ctx = lazy_loader.LazyLoader( variable 3488 ag_ctx = autograph_ctx.control_status_ctx()
|