Searched defs:TopK (Results 1 – 9 of 9) sorted by relevance
| /external/executorch/backends/qualcomm/builders/ |
| D | op_topk.py | 20 class TopK(NodeVisitor): class
|
| /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/aws-sdk-java-v2/services/bedrockagentruntime/src/main/resources/codegen-resources/ |
| D | service-2.json | 1002 "TopK":{ object
|
| /external/aws-sdk-java-v2/services/bedrockagent/src/main/resources/codegen-resources/ |
| D | service-2.json | 3116 "TopK":{ object
|
| /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
|