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 | 27 static void TopK(tensorflow::int64 batch_size, tensorflow::int64 input_size, in TopK() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 28 XlaOp TopK(XlaOp input, int64 k) { in TopK() function
|
| /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
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 476 TEST(TFCompileTest, TopK) { in TEST() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 24620 func TopK(scope *Scope, input tf.Output, k int64, optional ...TopKAttr) (values tf.Output, indices … func
|