| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| model_fn_test.py | 65 self, model_fn_ops, estimator_spec):
70 self.assertEqual(model_fn_ops.predictions, estimator_spec.predictions)
71 self.assertEqual(model_fn_ops.loss, estimator_spec.loss)
72 self.assertEqual(model_fn_ops.train_op, estimator_spec.train_op)
74 estimator_spec.eval_metric_ops)
76 estimator_spec.training_chief_hooks)
78 estimator_spec.training_hooks)
79 self.assertEqual(model_fn_ops.scaffold, estimator_spec.scaffold)
86 estimator_spec = model_fn_ops.estimator_spec()
[all...] |
| model_fn.py | 84 'tf.estimator.EstimatorSpec. You can use the `estimator_spec`'
212 def estimator_spec(self, default_serving_output_alternative_key=None): member in class:ModelFnOps
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| estimator_utils.py | 61 def estimator_spec_to_model_fn_ops(estimator_spec, export_alternatives=False):
64 estimator_spec.export_outputs)
69 mode=_core_mode_to_contrib_mode(estimator_spec.mode),
70 predictions=estimator_spec.predictions,
71 loss=estimator_spec.loss,
72 train_op=estimator_spec.train_op,
73 eval_metric_ops=estimator_spec.eval_metric_ops,
|
| model.py | 35 ESTIMATOR_SPEC = 1
180 elif output_type == ModelBuilderOutputType.ESTIMATOR_SPEC:
182 estimator_spec = head.create_estimator_spec(
189 estimator_spec = estimator_spec._replace(
190 training_hooks=training_hooks + list(estimator_spec.training_hooks))
191 return estimator_spec
400 elif output_type == ModelBuilderOutputType.ESTIMATOR_SPEC:
402 estimator_spec = head.create_estimator_spec(
409 estimator_spec = estimator_spec._replace
[all...] |
| /external/tensorflow/tensorflow/contrib/predictor/ |
| core_estimator_predictor.py | 35 estimator_spec = estimator.model_fn(
39 export_outputs = estimator_spec.export_outputs
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| xla.py | 618 estimator_spec = self._call_model_fn(features, labels,
622 captured_scaffold_fn.capture(estimator_spec.scaffold_fn)
631 with ops.control_dependencies([estimator_spec.train_op]):
632 return array_ops.identity(estimator_spec.loss)
643 estimator_spec = self._call_model_fn(features, labels,
647 captured_scaffold_fn.capture(estimator_spec.scaffold_fn)
654 if estimator_spec.eval_metrics:
655 (eval_metric_fn, eval_metric_fn_tensors) = estimator_spec.eval_metrics
669 return tuple([estimator_spec.loss] + eval_metric_fn_tensors)
692 def _verify_estimator_spec(self, estimator_spec)
[all...] |
| xla_test.py | 315 estimator_spec = model_fn(
319 self.assertEqual(estimator_spec.mode, _TRAIN)
322 self.assertEqual(sess.run(estimator_spec.loss), sess.run(loss))
323 self.assertEqual(sess.run(estimator_spec.train_op), sess.run(loss))
339 estimator_spec = model_fn(
343 self.assertEqual(estimator_spec.mode, _TRAIN)
346 self.assertEqual(sess.run(estimator_spec.loss), sess.run(loss))
347 self.assertEqual(sess.run(estimator_spec.train_op), sess.run(loss))
363 estimator_spec = model_fn_with_summary(
367 self.assertEqual(sess.run(estimator_spec.loss), _EXPECTED_LOSS
[all...] |
| /external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
| random_forest.py | 55 ESTIMATOR_SPEC = 1
295 estimator_spec = model_head.create_estimator_spec(
307 loss_op=estimator_spec.loss))
309 estimator_spec = estimator_spec._replace(
310 training_hooks=training_hooks + list(estimator_spec.training_hooks))
312 estimator_spec.predictions[keys_name] = keys
314 estimator_spec.predictions[TREE_PATHS_PREDICTION_KEY] = tree_paths
315 estimator_spec.predictions[VARIANCE_PREDICTION_KEY] = regression_variance
318 outputs = estimator_spec.export_output
[all...] |
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| tpu_gan_estimator_impl.py | 242 estimator_spec = _get_estimator_spec(
246 assert isinstance(estimator_spec, tpu_estimator.TPUEstimatorSpec)
247 return estimator_spec
279 estimator_spec = tpu_estimator.TPUEstimatorSpec(
293 estimator_spec = _get_eval_estimator_spec(
309 estimator_spec = _get_train_estimator_spec(
312 return estimator_spec
|
| gan_estimator_impl.py | 224 estimator_spec = model_fn_lib.EstimatorSpec(
233 estimator_spec = _get_eval_estimator_spec(
241 estimator_spec = _get_train_estimator_spec(
245 return estimator_spec
|
| stargan_estimator_impl.py | 216 estimator_spec = model_fn_lib.EstimatorSpec(
221 estimator_spec = _get_eval_estimator_spec(gan_model, gan_loss,
231 estimator_spec = _get_train_estimator_spec(gan_model, gan_loss, gopt,
234 return estimator_spec
|
| /external/tensorflow/tensorflow/python/tpu/ |
| tpu_estimator.py | [all...] |
