Home
last modified time | relevance | path

Searched refs:loss_callback (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/opt/python/training/
Dexternal_optimizer_test.py118 loss_callback = test.mock.Mock()
123 loss_callback=loss_callback,
128 loss_callback.assert_has_calls(loss_calls)
323 loss_callback = test.mock.Mock()
328 loss_callback=loss_callback,
336 loss_callback.assert_has_calls(loss_calls, any_order=True)
Dexternal_optimizer.py147 loss_callback=None, argument
174 loss_callback = loss_callback or (lambda *fetches: None)
180 feed_dict, fetches, loss_callback)