HomeSort by relevance Sort by last modified time
    Searched refs:captured_scaffold_fn (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/compiler/
xla.py 545 def _get_scaffold(captured_scaffold_fn):
546 """Retrieves the Scaffold from `captured_scaffold_fn`."""
547 scaffold_fn = captured_scaffold_fn.get()
581 train_step, captured_scaffold_fn = self._make_train_step(
588 scaffold=_get_scaffold(captured_scaffold_fn))
590 eval_step, captured_eval_metric_fn, captured_scaffold_fn = (
607 scaffold=_get_scaffold(captured_scaffold_fn))
614 captured_scaffold_fn = _CapturedObject()
622 captured_scaffold_fn.capture(estimator_spec.scaffold_fn)
624 captured_scaffold_fn.capture(None
    [all...]
  /external/tensorflow/tensorflow/python/tpu/
tpu_estimator.py     [all...]

Completed in 230 milliseconds