Home
last modified time | relevance | path

Searched refs:training_hooks (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dmodel.py145 training_hooks = []
151 training_hooks.append(
178 model_fn_ops.training_hooks.extend(training_hooks)
190 training_hooks=training_hooks + list(estimator_spec.training_hooks))
363 training_hooks = []
370 training_hooks.append(
397 model_fn_ops.training_hooks.extend(training_hooks)
410 training_hooks=training_hooks + list(estimator_spec.training_hooks))
Ddnn_tree_combined_estimator.py347 model_fn_ops.training_hooks.extend([
378 training_hooks = [
385 fusion_spec = fusion_spec._replace(training_hooks=training_hooks +
386 list(fusion_spec.training_hooks))
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dmodel_fn.py94 training_hooks=None, argument
192 if training_hooks is None:
193 training_hooks = []
194 for hook in training_hooks + training_chief_hooks:
208 training_hooks=training_hooks,
306 training_hooks=self.training_hooks,
Dkmeans.py140 training_hooks = [_InitializeClustersHook(
144 training_hooks.append(_LossRelativeChangeHook(relative_tolerance))
151 training_hooks=training_hooks)
Dmodel_fn_test.py60 training_hooks=[basic_session_run_hooks.StepCounterHook()],
77 self.assertAllEqual(model_fn_ops.training_hooks,
78 estimator_spec.training_hooks)
Destimator.py1071 for h in (all_hooks + model_fn_ops.training_hooks + chief_hooks +
1087 hooks=all_hooks + model_fn_ops.training_hooks,
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py220 training_hooks = []
255 training_hooks = [TensorForestRunOpAtEndHook(ops_at_end)]
267 training_hooks.append(
273 model_ops.training_hooks.extend(training_hooks)
303 training_hooks.append(
310 training_hooks=training_hooks + list(estimator_spec.training_hooks))
507 training_hooks = []
509 training_hooks += mops.training_hooks
536 training_hooks=training_hooks,
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals.py305 training_hooks = [sweep_hook, global_step_hook]
307 training_hooks.append(_StopAtSweepHook(max_sweeps))
315 training_hooks=training_hooks)
347 training_hooks=[])
369 training_hooks=[])
Dkmeans.py215 training_hooks = [
219 training_hooks.append(
240 training_hooks=training_hooks,
Dgmm.py169 training_hooks = [_InitializeClustersHook(
181 training_hooks=training_hooks)
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dhead_impl.py240 training_hooks = self._get_hooks_fn(train_ops)
245 training_hooks=training_hooks)
Dgan_estimator_impl.py333 training_hooks = get_hooks_fn(train_ops)
338 training_hooks=training_hooks)
Dstargan_estimator_impl.py337 training_hooks = get_hooks_fn(train_ops)
342 training_hooks=training_hooks)
Dgan_estimator_test.py167 self.assertIsNotNone(spec.training_hooks)
195 self.assertLen(spec.training_hooks, 4)
197 hook._sync_optimizer for hook in spec.training_hooks if
Dstargan_estimator_test.py189 self.assertIsNotNone(spec.training_hooks)
Dtpu_gan_estimator_test.py134 self.assertIsNotNone(spec.training_hooks)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.estimator.-estimator-spec.pbtxt47 name: "training_hooks"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.estimator.-estimator-spec.pbtxt47 name: "training_hooks"
/external/tensorflow/tensorflow/python/tpu/
Dtpu_estimator.py311 training_hooks=None, argument
322 training_hooks = tuple(training_hooks or [])
326 for hook in training_hooks + evaluation_hooks + prediction_hooks:
341 training_hooks=training_hooks,
375 training_hooks=self.training_hooks + hooks,
1471 captured_training_hooks.capture(estimator_spec.training_hooks)
2810 training_hooks=estimator_spec.training_hooks + (examples_hook,))
2838 compile_op, loss, host_call, scaffold, training_hooks = (
2915 if training_hooks:
2916 hooks.extend(training_hooks)
[all …]