Home
last modified time | relevance | path

Searched refs:in_top_k (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Din_topk_op_test.py34 precision = nn_ops.in_top_k(predictions, target, k)
71 nn_ops.in_top_k(predictions, target, 2).eval()
79 precision = nn_ops.in_top_k(predictions, target, k)
/external/tensorflow/tensorflow/examples/tutorials/mnist/
Dmnist.py145 correct = tf.nn.in_top_k(logits, labels, 1)
/external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/
Dcifar10_eval.py116 top_k_op = tf.nn.in_top_k(logits, labels, 1)
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Deval_metrics.py42 return metrics.mean(nn.in_top_k(probabilities, targets, k))
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dmain_estimator_tpu.py92 in_top_5 = tf.cast(tf.nn.in_top_k(logits, labels, 5), tf.float32)
/external/tensorflow/tensorflow/contrib/keras/api/keras/backend/
D__init__.py77 from tensorflow.python.keras.backend import in_top_k
/external/tensorflow/tensorflow/contrib/metrics/python/ops/
Dmetric_ops.py2167 in_top_k = math_ops.cast(nn.in_top_k(predictions, labels, k), dtypes.float32)
2168 return streaming_mean(in_top_k, weights, metrics_collections,
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py4666 def in_top_k(predictions, targets, k, name=None): function
4701 return in_top_k(predictions, targets, k, name)
4704 in_top_k_v2.__doc__ = in_top_k.__doc__
/external/tensorflow/tensorflow/python/keras/
Dmetrics.py2735 nn.in_top_k(y_pred, math_ops.argmax(y_true, axis=-1), k), axis=-1)
2747 return K.mean(nn.in_top_k(y_pred, math_ops.cast(y_true, 'int32'), k), axis=-1)
Dbackend.py4103 def in_top_k(predictions, targets, k): function
4116 return nn.in_top_k(predictions, targets, k)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt172 name: "in_top_k"
Dtensorflow.math.pbtxt176 name: "in_top_k"
Dtensorflow.keras.backend.pbtxt232 name: "in_top_k"
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/g3doc/
DREADME.md241 correct = tf.nn.in_top_k(output_class, y, 1)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt176 name: "in_top_k"
Dtensorflow.nn.pbtxt216 name: "in_top_k"
Dtensorflow.keras.backend.pbtxt236 name: "in_top_k"