Searched defs:topK (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/android/test_app/app/src/main/java/org/pytorch/testapp/ |
| D | Utils.java | 7 public static int[] topK(float[] a, final int topk) { in topK() method in Utils
|
| /external/guava/android/guava/src/com/google/common/collect/ |
| D | TopKSelector.java | 275 public List<T> topK() { in topK() method in TopKSelector
285 T[] topK = Arrays.copyOf(castBuffer, bufferSize); in topK() local
|
| /external/guava/guava/src/com/google/common/collect/ |
| D | TopKSelector.java | 276 public List<T> topK() { in topK() method in TopKSelector
286 T[] topK = Arrays.copyOf(castBuffer, bufferSize); in topK() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | TensorTopK.cu | 48 at::cuda::detail::TensorInfo<T, IndexType> topK, in gatherTopK()
190 at::cuda::detail::TensorInfo<T, IndexType> topK, in launch()
651 at::cuda::detail::TensorInfo<T, IndexType> topK, in launch()
|
| D | SortingRadixSelect.cuh | 313 scalar_t* topK) { in radixSelect()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding_test.cc | 641 auto topK = in TEST_F() local
|
| /external/aws-sdk-java-v2/services/bedrockagentruntime/src/main/resources/codegen-resources/ |
| D | service-2.json | 265 "topK":{"shape":"TopK"}, object
|
| /external/aws-sdk-java-v2/services/bedrockagent/src/main/resources/codegen-resources/ |
| D | service-2.json | 2062 "topK":{"shape":"TopK"}, object
|