Home
last modified time | relevance | path

Searched refs:loss_collection (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.losses.pbtxt9 …argspec: "args=[\'labels\', \'predictions\', \'weights\', \'scope\', \'loss_collection\', \'reduct…
13 …argspec: "args=[\'loss\', \'loss_collection\'], varargs=None, keywords=None, defaults=[\'losses\']…
17 …argspec: "args=[\'losses\', \'weights\', \'scope\', \'loss_collection\', \'reduction\'], varargs=N…
21 …gs=[\'labels\', \'predictions\', \'axis\', \'weights\', \'scope\', \'loss_collection\', \'reductio…
25 …argspec: "args=[\'scope\', \'loss_collection\'], varargs=None, keywords=None, defaults=[\'None\', …
41 …argspec: "args=[\'labels\', \'logits\', \'weights\', \'scope\', \'loss_collection\', \'reduction\'…
45 …s=[\'labels\', \'predictions\', \'weights\', \'delta\', \'scope\', \'loss_collection\', \'reductio…
49 …[\'labels\', \'predictions\', \'weights\', \'epsilon\', \'scope\', \'loss_collection\', \'reductio…
53 …argspec: "args=[\'labels\', \'predictions\', \'weights\', \'scope\', \'loss_collection\'], varargs…
57 …argspec: "args=[\'labels\', \'predictions\', \'weights\', \'scope\', \'loss_collection\', \'reduct…
[all …]
/external/tensorflow/tensorflow/python/ops/losses/
Dlosses_impl.py142 losses, weights=1.0, scope=None, loss_collection=ops.GraphKeys.LOSSES, argument
204 util.add_loss(loss, loss_collection)
212 loss_collection=ops.GraphKeys.LOSSES, argument
259 losses, weights, scope, loss_collection, reduction=reduction)
267 loss_collection=ops.GraphKeys.LOSSES, argument
316 losses, weights, scope, loss_collection, reduction=reduction)
322 loss_collection=ops.GraphKeys.LOSSES, argument
367 losses, weights, scope, loss_collection, reduction=reduction)
373 loss_collection=ops.GraphKeys.LOSSES, argument
444 losses, weights, scope, loss_collection, reduction=reduction)
[all …]
Dutil.py178 def add_loss(loss, loss_collection=ops.GraphKeys.LOSSES): argument
188 if loss_collection and not context.executing_eagerly():
189 ops.add_to_collection(loss_collection, loss)
193 def get_losses(scope=None, loss_collection=ops.GraphKeys.LOSSES): argument
203 return ops.get_collection(loss_collection, scope)
/external/tensorflow/tensorflow/python/kernel_tests/
Dlosses_test.py1351 losses.absolute_difference(logits, labels, loss_collection=None)
1352 losses.log_loss(logits, labels, loss_collection=None)
1353 losses.mean_squared_error(logits, labels, loss_collection=None)
1354 losses.sigmoid_cross_entropy(logits, labels, loss_collection=None)
1355 losses.softmax_cross_entropy(logits, labels, loss_collection=None)