Searched refs:training_chief_hooks (Results 1 – 8 of 8) sorted by relevance
93 training_chief_hooks=None, argument190 if training_chief_hooks is None:191 training_chief_hooks = []194 for hook in training_hooks + training_chief_hooks:207 training_chief_hooks=training_chief_hooks,305 training_chief_hooks=self.training_chief_hooks,
59 training_chief_hooks=[basic_session_run_hooks.StepCounterHook()],75 self.assertAllEqual(model_fn_ops.training_chief_hooks,76 estimator_spec.training_chief_hooks)
279 training_chief_hooks=(model_fn_ops.training_chief_hooks +
1072 model_fn_ops.training_chief_hooks)1088 chief_only_hooks=chief_hooks + model_fn_ops.training_chief_hooks,
43 name: "training_chief_hooks"
175 return model_fn_ops._replace(training_chief_hooks=(176 model_fn_ops.training_chief_hooks + [update_weights_hook]))
1722 if estimator_spec.training_chief_hooks:2945 training_chief_hooks=chief_hooks,