Searched refs:use_callable_loss (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | minimize_loss_test.py | 77 combinations.combine(mode=["graph"], use_callable_loss=[True, False])
78 + combinations.combine(mode=["eager"], use_callable_loss=[True])) +
82 mode=["graph", "eager"], use_callable_loss=[True])) +
87 use_callable_loss=[True]) + combinations.combine(
91 use_callable_loss=[True, False]))
92 def testTrainNetwork(self, distribution, optimizer_fn, use_callable_loss): argument
96 optimizer, use_bias=True, use_callable_loss=use_callable_loss)
128 combinations.combine(mode=["graph"], use_callable_loss=[True, False])
129 + combinations.combine(mode=["eager"], use_callable_loss=[True])) +
133 mode=["graph", "eager"], use_callable_loss=[True])))
[all …]
|
| D | test_example.py | 31 def minimize_loss_example(optimizer, use_bias=False, use_callable_loss=True): argument
51 elif use_callable_loss:
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | single_loss_example.py | 54 def minimize_loss_example(optimizer, use_bias=False, use_callable_loss=True): argument
74 elif use_callable_loss:
|