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

  /external/tensorflow/tensorflow/contrib/distribute/python/examples/
simple_estimator_example.py 43 return tf.estimator.EstimatorSpec(mode, predictions=predictions)
52 return tf.estimator.EstimatorSpec(
59 return tf.estimator.EstimatorSpec(mode, loss=loss_fn(), train_op=train_op)
  /external/tensorflow/tensorflow/examples/get_started/regression/
custom_regression.py 48 return tf.estimator.EstimatorSpec(
65 return tf.estimator.EstimatorSpec(
77 return tf.estimator.EstimatorSpec(
  /external/tensorflow/tensorflow/examples/learn/
iris_custom_decay_dnn.py 47 return tf.estimator.EstimatorSpec(mode, predictions=predictions)
60 return tf.estimator.EstimatorSpec(mode, loss=loss, train_op=train_op)
67 return tf.estimator.EstimatorSpec(
iris_custom_model.py 48 return tf.estimator.EstimatorSpec(mode, predictions=predictions)
57 return tf.estimator.EstimatorSpec(mode, loss=loss, train_op=train_op)
64 return tf.estimator.EstimatorSpec(
  /external/tensorflow/tensorflow/examples/tutorials/layers/
cnn_mnist.py 97 return tf.estimator.EstimatorSpec(mode=mode, predictions=predictions)
108 return tf.estimator.EstimatorSpec(mode=mode, loss=loss, train_op=train_op)
114 return tf.estimator.EstimatorSpec(
  /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
main_estimator.py 40 An instance of `tf.estimator.EstimatorSpec`
62 return tf.estimator.EstimatorSpec(mode=mode, loss=loss, train_op=train_op)
70 return tf.estimator.EstimatorSpec(
84 return tf.estimator.EstimatorSpec(
  /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
head_impl.py 65 passed into `tf.estimator.EstimatorSpec` during evaluation.
110 passed into `tf.estimator.EstimatorSpec` during evaluation.
175 """Returns `EstimatorSpec` that a model_fn can return.
190 `EstimatorSpec`.
203 return model_fn_lib.EstimatorSpec(
228 return model_fn_lib.EstimatorSpec(
241 return model_fn_lib.EstimatorSpec(
gan_estimator_impl.py 151 passed into `tf.estimator.EstimatorSpec` during evaluation.
188 # Make the EstimatorSpec, which incorporates the GANModel, losses, eval
222 """Get the EstimatorSpec for the current mode."""
224 estimator_spec = model_fn_lib.EstimatorSpec(
300 """Return an EstimatorSpec for the eval case."""
319 return model_fn_lib.EstimatorSpec(
329 """Return an EstimatorSpec for the train case."""
334 return model_fn_lib.EstimatorSpec(
stargan_estimator_impl.py 140 passed into `tf.estimator.EstimatorSpec` during evaluation.
178 # Make the EstimatorSpec, which incorporates the StarGANModel, losses,
214 """Get the EstimatorSpec for the current mode."""
216 estimator_spec = model_fn_lib.EstimatorSpec(
300 """Return an EstimatorSpec for the eval case."""
320 return model_fn_lib.EstimatorSpec(
333 """Return an EstimatorSpec for the train case."""
338 return model_fn_lib.EstimatorSpec(
latent_gan_estimator_impl.py 131 return model_fn_lib.EstimatorSpec(mode=mode,
  /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
head.py 107 return estimator_lib.EstimatorSpec(
129 return estimator_lib.EstimatorSpec(
142 return estimator_lib.EstimatorSpec(
160 return estimator_lib.EstimatorSpec(
233 """Performs basic error checking and returns an EstimatorSpec."""
359 return estimator_lib.EstimatorSpec(
  /external/tensorflow/tensorflow/python/compiler/tensorrt/test/
quantization_mnist_test.py 28 from tensorflow.python.estimator.model_fn import EstimatorSpec
236 return EstimatorSpec(
241 return EstimatorSpec(mode, loss=loss, train_op=train_op)
  /external/tensorflow/tensorflow/contrib/predictor/
testing_common.py 61 return model_fn.EstimatorSpec(mode=mode,
  /external/tensorflow/tensorflow/contrib/compiler/
xla_test.py 275 return model_fn_lib.EstimatorSpec(
359 return model_fn_lib.EstimatorSpec(
405 return model_fn_lib.EstimatorSpec(
413 ValueError, 'EstimatorSpec.eval_metric_ops is not supported with XLA '
422 return model_fn_lib.EstimatorSpec(
xla.py 584 return model_fn_lib.EstimatorSpec(
603 return model_fn_lib.EstimatorSpec(
698 logging.warning('EstimatorSpec.scaffold is ignored with XLA compilation'
705 raise ValueError('EstimatorSpec.eval_metric_ops is not supported with '
761 return EstimatorSpec(...)
773 return EstimatorSpec(...)
  /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
model_fn.py 84 'tf.estimator.EstimatorSpec. You can use the `estimator_spec`'
213 """Creates an equivalent `EstimatorSpec`.
218 multiple heads), `EstimatorSpec` requires a default head that will be
227 Instance of `EstimatorSpec` that is equivalent to this `ModelFnOps`
298 return core_model_fn_lib.EstimatorSpec(
  /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
checkpoint_input_pipeline_hook_test.py 52 return model_fn.EstimatorSpec(
  /external/tensorflow/tensorflow/contrib/factorization/python/ops/
kmeans.py 166 A `tf.estimator.EstimatorSpec` (see `tf.estimator.Estimator`) specifying
231 return model_fn_lib.EstimatorSpec(
  /external/tensorflow/tensorflow/contrib/distribute/python/
values_test.py 320 spec = model_fn_lib.EstimatorSpec(
332 isinstance(merged_estimator_spec, model_fn_lib.EstimatorSpec))
340 # Scaffold is populated by `EstimatorSpec.__new__`.
  /external/tensorflow/tensorflow/python/tpu/
tpu_estimator.py 269 See `EstimatorSpec` for `mode`, `predictions`, `loss`, `train_op`, and
276 `tf.estimator.Estimator`. While `EstimatorSpec.eval_metric_ops` expects a
346 """Creates an equivalent `EstimatorSpec` used by CPU train/eval."""
367 return model_fn_lib.EstimatorSpec(
    [all...]

Completed in 848 milliseconds