Searched refs:step_callback (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/contrib/opt/python/training/ |
D | external_optimizer_test.py | 44 def _minimize(self, initial_val, loss_grad_func, step_callback, argument 50 step_callback(initial_val) 117 step_callback = test.mock.Mock() 124 step_callback=step_callback) 130 args, _ = step_callback.call_args 322 step_callback = test.mock.Mock() 329 step_callback=step_callback) 338 args, _ = step_callback.call_args
|
D | external_optimizer.py | 146 step_callback=None, argument 175 step_callback = step_callback or (lambda xk: None) 206 step_callback=step_callback, 220 packed_bounds, step_callback, optimizer_kwargs): argument 363 packed_bounds, step_callback, optimizer_kwargs): argument 382 'callback': step_callback,
|