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

  /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
tpu_gan_estimator_impl.py 240 # Make the TPUEstimatorSpec, which incorporates the GANModel, losses, eval
246 assert isinstance(estimator_spec, tpu_estimator.TPUEstimatorSpec)
277 """Get the TPUEstimatorSpec for the current mode."""
279 estimator_spec = tpu_estimator.TPUEstimatorSpec(
317 """Return an TPUEstimatorSpec for the eval case."""
357 return tpu_estimator.TPUEstimatorSpec(
367 """Return a TPUEstimatorSpec for the train case."""
396 return tpu_estimator.TPUEstimatorSpec(
tpu_gan_estimator_test.py 126 self.assertIsInstance(spec, tpu_estimator.TPUEstimatorSpec)
  /external/tensorflow/tensorflow/contrib/compiler/
xla_test.py 414 'compilation. Please use TPUEstimatorSpec.eval_metrics instead.'):
436 return tpu_estimator.TPUEstimatorSpec(
455 return tpu_estimator.TPUEstimatorSpec(
464 'element of TPUEstimatorSpec.eval_metrics) but they are not '
466 'TPUEstimatorSpec.eval_metrics).')):
479 return tpu_estimator.TPUEstimatorSpec(
488 'tensors (second element of TPUEstimatorSpec.eval_metrics) '
490 'TPUEstimatorSpec.eval_metrics).')):
499 return tpu_estimator.TPUEstimatorSpec(
515 return tpu_estimator.TPUEstimatorSpec(
    [all...]
  /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
main_estimator_tpu.py 44 https://www.tensorflow.org/api_docs/python/tf/contrib/tpu/TPUEstimatorSpec
77 https://www.tensorflow.org/api_docs/python/tf/contrib/tpu/TPUEstimatorSpec
111 An instance of `tf.contrib.tpu.TPUEstimatorSpec`
156 return tf.contrib.tpu.TPUEstimatorSpec(
163 return tf.contrib.tpu.TPUEstimatorSpec(
173 return tf.contrib.tpu.TPUEstimatorSpec(
  /external/tensorflow/tensorflow/python/tpu/
tpu_estimator.py 266 class TPUEstimatorSpec(model_fn_lib._TPUEstimatorSpec): # pylint: disable=protected-access
277 dict, `TPUEstimatorSpec.eval_metrics` is a tuple of `metric_fn` and `tensors`.
314 """Creates a validated `TPUEstimatorSpec` instance."""
331 return super(TPUEstimatorSpec, cls).__new__(
    [all...]

Completed in 771 milliseconds