Searched defs:TopK (Results 1 – 6 of 6) sorted by relevance
| /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/compiler/xla/service/cpu/ |
| D | runtime_topk.cc | 28 static void TopK(int64_t batch_size, int64_t input_size, int64_t k, in TopK() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 28 XlaOp TopK(XlaOp input, int64_t k) { in TopK() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | topk_v2.cc | 177 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 | 493 TEST(TFCompileTest, TopK) { in TEST() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 52987 func TopK(scope *Scope, input tf.Output, k int64, optional ...TopKAttr) (values tf.Output, indices … func
|