| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting_test.cc | 35 xla::GetTupleElement(xla::TopK(x, 3), 0); in XLA_TEST_F()
43 xla::GetTupleElement(xla::TopK(x_rev, 3), 1); in XLA_TEST_F()
52 xla::GetTupleElement(xla::TopK(x_rev, 3), 1); in XLA_TEST_F()
65 xla::GetTupleElement(xla::TopK(x, kSize), 0); in XLA_TEST_F()
75 xla::GetTupleElement(xla::TopK(a, 5), 1); in XLA_TEST_F()
129 xla::GetTupleElement(xla::TopK(x, 1000), 0); in XLA_TEST_F()
|
| D | sorting.h | 27 XlaOp TopK(XlaOp input, int64 k);
|
| D | sorting.cc | 28 XlaOp TopK(XlaOp input, int64 k) { in TopK() function
87 return TopK(input, k); in TopKWithPartitions()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TopK.pbtxt | 2 name: "TopK"
49 name: "TopK"
98 name: "TopK"
148 name: "TopK"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TopK.pbtxt | 2 name: "TopK"
49 name: "TopK"
98 name: "TopK"
148 name: "TopK"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | topk_op.cc | 40 class TopK : public OpKernel { class
42 explicit TopK(OpKernelConstruction* context) : OpKernel(context) { in TopK() function in tensorflow::TopK
237 TopK<CPUDevice, type>)
240 REGISTER_KERNELS_NAME(TopK, type); \
269 TopK<GPUDevice, type>) \
274 TopK<GPUDevice, type>)
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | topk_v2.cc | 176 void TopK(int32 row_size, int32 num_rows, const T* data, int32 k, in TopK() function
255 TopK(row_size, num_rows, GetTensorData<float>(input), k, in Eval()
259 TopK(row_size, num_rows, input->data.uint8, k, output_indexes->data.i32, in Eval()
263 TopK(row_size, num_rows, input->data.int8, k, output_indexes->data.i32, in Eval()
267 TopK(row_size, num_rows, input->data.i32, k, output_indexes->data.i32, in Eval()
271 TopK(row_size, num_rows, input->data.i64, k, output_indexes->data.i32, in Eval()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_topk_test.cc | 34 TopK(input, 10); in TEST_F()
61 TopK(input, 10); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_topk.cc | 27 static void TopK(tensorflow::int64 batch_size, tensorflow::int64 input_size, in TopK() function
75 TopK(batch_size, input_size, k, values, out_values, out_indices); in __xla_cpu_runtime_TopKF32()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TopKUnique.pbtxt | 3 summary: "Returns the TopK unique values in the array in sorted order."
12 returned. If a TopK element never appears in the input due to padding
|
| D | api_def_TopKWithUnique.pbtxt | 3 summary: "Returns the TopK values in the array in sorted order."
|
| D | api_def_InTopKV2.pbtxt | 32 from the `TopK` op in its handling of ties; if multiple classes have the
|
| D | api_def_InTopK.pbtxt | 32 from the `TopK` op in its handling of ties; if multiple classes have the
|
| D | api_def_TopKV2.pbtxt | 4 name: "TopK"
|
| D | api_def_TopK.pbtxt | 2 graph_op_name: "TopK"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_TopK.pbtxt | 2 graph_op_name: "TopK"
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_TopK.pbtxt | 2 graph_op_name: "TopK"
|
| D | api_def_TopKV2.pbtxt | 4 name: "nn.TopK"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | topk_op.cc | 51 xla::XlaOp output_tuple = TopK(context->Input(0), k); in Compile()
|
| D | image_ops.cc | 545 xla::XlaOp output_tuple = TopK(scores_included, output_size); in Compile()
|
| /external/tensorflow/tensorflow/examples/label_image/ |
| D | main.cc | 212 TopK(root.WithOpName(output_name), outputs[0], how_many_labels); in GetTopLabels()
|
| /external/tensorflow/tensorflow/examples/multibox_detector/ |
| D | main.cc | 193 TopK(root.WithOpName(output_name), outputs[0], how_many_labels); in GetTopDetections()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding_test.cc | 642 ops::TopK(s.WithOpName("topK"), c0, k, ops::TopK::Sorted(false)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 290 ops.def("TopK", &TopK, py::arg("input"), py::arg("k")); in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 646 * `raw_ops.TopK`
|