Searched refs:InTopK (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | in_topk_op.cc | 32 class InTopK : public OpKernel { class
34 explicit InTopK(OpKernelConstruction* context) : OpKernel(context) { in InTopK() function in tensorflow::InTopK
93 InTopK<CPUDevice, float, int32>);
100 InTopK<CPUDevice, float, int64>);
109 InTopK<CPUDevice, float, int32>);
117 InTopK<CPUDevice, float, int64>);
140 InTopK<GPUDevice, float, int32>);
143 InTopK<GPUDevice, float, int64>);
|
| D | in_topk_op_test.cc | 36 static Graph* InTopK(int num_targets, int num_classes, T top_k) { in InTopK() function
71 test::Benchmark(#DEVICE, InTopK<T>(TARGETS, CLASSES, K), \
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_InTopK.pbtxt | 2 graph_op_name: "InTopK"
31 all predictions for example `i`. Note that the behavior of `InTopK` differs
|
| D | api_def_InTopKV2.pbtxt | 31 all predictions for example `i`. Note that the behavior of `InTopK` differs
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_InTopK.pbtxt | 2 graph_op_name: "InTopK"
|
| D | api_def_InTopKV2.pbtxt | 4 name: "nn.InTopK"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_InTopK.pbtxt | 2 graph_op_name: "InTopK"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | InTopK.pbtxt | 2 name: "InTopK"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | InTopK.pbtxt | 2 name: "InTopK"
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 220 * `raw_ops.InTopK`
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1916 name: "InTopK"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1916 name: "InTopK"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 14237 name: "InTopK"
14256 name: "InTopK"
|
| D | ops_history.v2.pbtxt | 29390 name: "InTopK"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 44073 func InTopK(scope *Scope, predictions tf.Output, targets tf.Output, k int64) (precision tf.Output) { func
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 5485 all predictions for example `i`. Note that the behavior of `InTopK` differs
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 19546 name: "InTopK"
|