Searched refs:TOPK_V2 (Results 1 – 17 of 17) sorted by relevance
/frameworks/ml/nn/runtime/test/generated/spec_V1_2/ |
D | topk_v2.example.cpp | 60 .type = TestOperationType::TOPK_V2 in get_test_model() 161 .type = TestOperationType::TOPK_V2 in get_test_model_all_inputs_as_internal() 228 .type = TestOperationType::TOPK_V2 in get_test_model_relaxed() 329 .type = TestOperationType::TOPK_V2 in get_test_model_relaxed_all_inputs_as_internal() 396 .type = TestOperationType::TOPK_V2 in get_test_model_float16() 497 .type = TestOperationType::TOPK_V2 in get_test_model_float16_all_inputs_as_internal() 564 .type = TestOperationType::TOPK_V2 in get_test_model_2() 665 .type = TestOperationType::TOPK_V2 in get_test_model_all_inputs_as_internal_2() 732 .type = TestOperationType::TOPK_V2 in get_test_model_relaxed_2() 833 .type = TestOperationType::TOPK_V2 in get_test_model_relaxed_all_inputs_as_internal_2() [all …]
|
/frameworks/ml/nn/runtime/test/generated/spec_V1_3/ |
D | topk_v2_quant8_signed.example.cpp | 60 .type = TestOperationType::TOPK_V2 in get_test_model() 161 .type = TestOperationType::TOPK_V2 in get_test_model_all_inputs_as_internal()
|
/frameworks/ml/nn/runtime/test/android_fuzzing/corpus/ |
D | seed048 | 73 type: TOPK_V2
|
D | seed377 | 76 type: TOPK_V2
|
D | seed218 | 76 type: TOPK_V2
|
D | seed334 | 76 type: TOPK_V2
|
D | seed165 | 93 type: TOPK_V2
|
/frameworks/ml/nn/runtime/test/android_fuzzing/ |
D | StaticAssert.cpp | 160 static_assert(static_cast<TestOperationType>(TOPK_V2) == TestOperationType::TOPK_V2);
|
D | Model.proto | 130 TOPK_V2 = 90; enumerator
|
/frameworks/ml/nn/common/operations/ |
D | TopK_V2.cpp | 143 NN_REGISTER_OPERATION(TOPK_V2, "TOPK_V2", topk_v2::validate, topk_v2::prepare, topk_v2::execute);
|
/frameworks/ml/nn/runtime/test/ |
D | TestAssertions.cpp | 133 CHECK_TEST_ENUM(TestOperationType, TOPK_V2);
|
D | TestValidateOperations.cpp | 1283 TEST(OperationValidationTest, TOPK_V2) { in TEST() argument
|
/frameworks/ml/nn/tools/test_generator/test_harness/include/ |
D | TestHarness.h | 185 TOPK_V2 = 90, enumerator
|
/frameworks/ml/nn/runtime/test/fuzzing/ |
D | TestRandomGraph.cpp | 668 TEST_SINGLE_OPERATION(TOPK_V2, V1_2, kStrictCriteria); 735 TEST_SINGLE_OPERATION(TOPK_V2, V1_3, kStrictCriteria);
|
/frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
D | Selection.cpp | 213 .opType = TestOperationType::TOPK_V2, \
|
/frameworks/ml/nn/runtime/ |
D | NeuralNetworks.cpp | 519 static_assert(static_cast<int32_t>(OperationType::TOPK_V2) == ANEURALNETWORKS_TOPK_V2,
|
/frameworks/ml/nn/tools/api/ |
D | types.spec | 5567 %{DeclareOperation_1.2 TOPK_V2 90},
|