| /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()
53 xla::GetTupleElement(xla::TopK(x_rev, 3), 1); in XLA_TEST_F()
62 xla::GetTupleElement(xla::TopK(x_rev, 3), 1); in XLA_TEST_F()
78 xla::GetTupleElement(xla::TopK(x, kSize), 0); in XLA_TEST_F()
88 xla::GetTupleElement(xla::TopK(a, 5), 1); in XLA_TEST_F()
|
| D | sorting.h | 27 XlaOp TopK(XlaOp input, int64 k);
|
| D | sorting.cc | 24 XlaOp TopK(XlaOp input, int64 k) { in TopK() function
|
| /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
235 TopK<CPUDevice, type>)
238 REGISTER_KERNELS_NAME(TopK, type); \
267 TopK<GPUDevice, type>) \
272 TopK<GPUDevice, type>)
|
| D | nn_ops_test.cc | 1313 auto top_k = ops::TopK(root, input, input_k, ops::TopK::Sorted(true)); in BM_TopK()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | topk_v2.cc | 138 void TopK(int32 row_size, int32 num_rows, const T* data, int32 k, in TopK() function
203 TopK(row_size, num_rows, input->data.f, k, output_indexes->data.i32, in Eval()
207 TopK(row_size, num_rows, input->data.uint8, k, output_indexes->data.i32, in Eval()
211 TopK(row_size, num_rows, input->data.int8, k, output_indexes->data.i32, in Eval()
215 TopK(row_size, num_rows, input->data.i32, k, output_indexes->data.i32, in Eval()
219 TopK(row_size, num_rows, input->data.i64, k, output_indexes->data.i32, in Eval()
|
| /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 | 539 xla::XlaOp output_tuple = TopK(scores_included, output_size); in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| 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_TopK.pbtxt | 2 graph_op_name: "TopK"
|
| D | api_def_TopKV2.pbtxt | 4 name: "TopK"
|
| /external/tensorflow/tensorflow/examples/label_image/ |
| D | main.cc | 211 TopK(root.WithOpName(output_name), outputs[0], how_many_labels); in GetTopLabels()
|
| /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/examples/multibox_detector/ |
| D | main.cc | 189 TopK(root.WithOpName(output_name), outputs[0], how_many_labels); in GetTopDetections()
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 454 TEST(TFCompileTest, TopK) { in TEST() argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3385 auto topk = ops::TopK(s.WithOpName("my_topk"), input, weights); in TEST_F()
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 39229 name: "TopK"
39265 name: "TopK"
39308 name: "TopK"
39350 name: "TopK"
39393 name: "TopK"
39437 name: "TopK"
|
| D | ops_history.v1.pbtxt | 85808 name: "TopK"
85855 name: "TopK"
85904 name: "TopK"
85954 name: "TopK"
|
| D | ops_history.v2.pbtxt | 85181 name: "TopK"
85228 name: "TopK"
85277 name: "TopK"
85327 name: "TopK"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4152 name: "TopK"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4152 name: "TopK"
|