Searched refs:ARGMAX (Results 1 – 17 of 17) sorted by relevance
| /frameworks/ml/nn/runtime/test/generated/spec_V1_2/ |
| D | argmax_2.example.cpp | 50 .type = TestOperationType::ARGMAX in get_test_model()
141 .type = TestOperationType::ARGMAX in get_test_model_all_inputs_as_internal()
198 .type = TestOperationType::ARGMAX in get_test_model_relaxed()
289 .type = TestOperationType::ARGMAX in get_test_model_relaxed_all_inputs_as_internal()
346 .type = TestOperationType::ARGMAX in get_test_model_float16()
437 .type = TestOperationType::ARGMAX in get_test_model_float16_all_inputs_as_internal()
494 .type = TestOperationType::ARGMAX in get_test_model_int32()
551 .type = TestOperationType::ARGMAX in get_test_model_quant8()
642 .type = TestOperationType::ARGMAX in get_test_model_quant8_all_inputs_as_internal()
|
| D | argmax_1.example.cpp | 50 .type = TestOperationType::ARGMAX in get_test_model()
141 .type = TestOperationType::ARGMAX in get_test_model_all_inputs_as_internal()
198 .type = TestOperationType::ARGMAX in get_test_model_relaxed()
289 .type = TestOperationType::ARGMAX in get_test_model_relaxed_all_inputs_as_internal()
346 .type = TestOperationType::ARGMAX in get_test_model_float16()
437 .type = TestOperationType::ARGMAX in get_test_model_float16_all_inputs_as_internal()
494 .type = TestOperationType::ARGMAX in get_test_model_int32()
551 .type = TestOperationType::ARGMAX in get_test_model_quant8()
642 .type = TestOperationType::ARGMAX in get_test_model_quant8_all_inputs_as_internal()
|
| D | argmax_3.example.cpp | 50 .type = TestOperationType::ARGMAX in get_test_model()
141 .type = TestOperationType::ARGMAX in get_test_model_all_inputs_as_internal()
198 .type = TestOperationType::ARGMAX in get_test_model_relaxed()
289 .type = TestOperationType::ARGMAX in get_test_model_relaxed_all_inputs_as_internal()
346 .type = TestOperationType::ARGMAX in get_test_model_float16()
437 .type = TestOperationType::ARGMAX in get_test_model_float16_all_inputs_as_internal()
494 .type = TestOperationType::ARGMAX in get_test_model_int32()
551 .type = TestOperationType::ARGMAX in get_test_model_quant8()
642 .type = TestOperationType::ARGMAX in get_test_model_quant8_all_inputs_as_internal()
|
| D | argmax_b155660285.example.cpp | 50 .type = TestOperationType::ARGMAX in get_test_model()
141 .type = TestOperationType::ARGMAX in get_test_model_all_inputs_as_internal()
|
| /frameworks/ml/nn/runtime/test/generated/spec_V1_3/ |
| D | argmax_quant8_signed.example.cpp | 50 .type = TestOperationType::ARGMAX in get_test_model_quant8_signed()
141 .type = TestOperationType::ARGMAX in get_test_model_quant8_signed_all_inputs_as_internal()
198 .type = TestOperationType::ARGMAX in get_test_model_quant8_signed_2()
289 .type = TestOperationType::ARGMAX in get_test_model_quant8_signed_all_inputs_as_internal_2()
346 .type = TestOperationType::ARGMAX in get_test_model_quant8_signed_3()
437 .type = TestOperationType::ARGMAX in get_test_model_quant8_signed_all_inputs_as_internal_3()
|
| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Reduce.cpp | 125 DEFINE_ARGMIN_MAX_SIGNATURE(ARGMAX, V1_2, TestOperandType::TENSOR_FLOAT32,
131 DEFINE_ARGMIN_MAX_SIGNATURE(ARGMAX, V1_3, TestOperandType::TENSOR_QUANT8_ASYMM_SIGNED);
|
| /frameworks/ml/nn/runtime/test/android_fuzzing/corpus/ |
| D | seed457 | 57 type: ARGMAX
|
| D | seed066 | 57 type: ARGMAX
|
| /frameworks/ml/nn/runtime/test/android_fuzzing/ |
| D | StaticAssert.cpp | 97 static_assert(static_cast<TestOperationType>(ARGMAX) == TestOperationType::ARGMAX);
|
| D | Model.proto | 79 ARGMAX = 39; enumerator
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestAssertions.cpp | 82 CHECK_TEST_ENUM(TestOperationType, ARGMAX);
|
| D | TestValidateOperations.cpp | 1056 TEST(OperationValidationTest, ARGMAX) { in TEST() argument
|
| /frameworks/ml/nn/tools/test_generator/test_harness/include/ |
| D | TestHarness.h | 134 ARGMAX = 39, enumerator
|
| /frameworks/ml/nn/runtime/test/fuzzing/ |
| D | TestRandomGraph.cpp | 634 TEST_SINGLE_OPERATION(ARGMAX, V1_2, kStrictCriteria);
708 TEST_SINGLE_OPERATION(ARGMAX, V1_3, kStrictCriteria);
|
| /frameworks/ml/nn/runtime/ |
| D | NeuralNetworks.cpp | 401 static_assert(static_cast<int32_t>(OperationType::ARGMAX) == ANEURALNETWORKS_ARGMAX,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1392 case OperationType::ARGMAX: in executeOperation()
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 3027 %{DeclareOperation_1.2 ARGMAX 39},
3054 %{DeclareOperation_1.2 ARGMIN 40}, // See ARGMAX for naming discussion.
|