Home
last modified time | relevance | path

Searched refs:sparse_softmax_cross_entropy (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dlosses_test.py235 losses.sparse_softmax_cross_entropy(labels, logits, weights=None)
243 loss = losses.sparse_softmax_cross_entropy(labels, logits)
252 losses.sparse_softmax_cross_entropy(labels, logits)
260 loss = losses.sparse_softmax_cross_entropy(labels, logits)
270 loss = losses.sparse_softmax_cross_entropy(labels, logits)
281 loss = losses.sparse_softmax_cross_entropy(labels, logits)
292 loss = losses.sparse_softmax_cross_entropy(labels, logits)
303 loss = losses.sparse_softmax_cross_entropy(labels, logits)
314 loss = losses.sparse_softmax_cross_entropy(labels, logits, weights)
324 loss = losses.sparse_softmax_cross_entropy(labels, logits,
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dasync_checkpoint_test.py66 loss = losses.sparse_softmax_cross_entropy(labels=labels, logits=logits)
87 loss = losses.sparse_softmax_cross_entropy(labels=labels, logits=logits)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.losses.pbtxt68 name: "sparse_softmax_cross_entropy"
/aosp_15_r20/external/tensorflow/tensorflow/examples/speech_commands/
H A Dtrain.py149 cross_entropy_mean = tf.compat.v1.losses.sparse_softmax_cross_entropy(
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
H A Dquantization_mnist_test.py236 loss = losses.sparse_softmax_cross_entropy(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/losses/
H A Dlosses_impl.py1047 def sparse_softmax_cross_entropy( function
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dlosses.py2160 losses_impl.sparse_softmax_cross_entropy: 'int32',
/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/
H A Dauto_mixed_precision_test.py876 loss = losses.sparse_softmax_cross_entropy(yt, y)