Searched refs:topk_v2 (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/runtime/test/generated/tests/ |
| D | topk_v2.mod.py.cpp | 5 namespace topk_v2 { namespace
12 TEST_F(GeneratedTests, topk_v2) { in TEST_F() argument
13 execute(topk_v2::CreateModel, in TEST_F()
14 topk_v2::is_ignored, in TEST_F()
15 topk_v2::get_examples()); in TEST_F()
17 TEST_AVAILABLE_SINCE(V1_2, topk_v2, topk_v2::CreateModel) in TEST_AVAILABLE_SINCE() argument
20 execute(topk_v2::CreateModel_relaxed, in TEST_AVAILABLE_SINCE()
21 topk_v2::is_ignored_relaxed, in TEST_AVAILABLE_SINCE()
22 topk_v2::get_examples_relaxed()); in TEST_AVAILABLE_SINCE()
26 execute(topk_v2::CreateModel_float16, in TEST_F()
[all …]
|
| /frameworks/ml/nn/common/operations/ |
| D | TopK_V2.h | 24 namespace topk_v2 {
|
| D | TopK_V2.cpp | 24 namespace topk_v2 { namespace
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1723 success = topk_v2::prepare(input.shape(), k, &valuesShape, &indicesShape) && in executeOperation()
1726 topk_v2::eval(input.buffer, input.shape(), k, values.buffer, valuesShape, in executeOperation()
|