/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...] |