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,
|