Searched defs:TopK (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| 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
|
| /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
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 454 TEST(TFCompileTest, TopK) { in TEST() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 22096 func TopK(scope *Scope, input tf.Output, k int64, optional ...TopKAttr) (values tf.Output, indices … func
|