Home
last modified time | relevance | path

Searched refs:ANEURALNETWORKS_ARGMAX (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/runtime/include/
DNeuralNetworks.h2536 ANEURALNETWORKS_ARGMAX = 39, enumerator
/frameworks/ml/nn/runtime/
DNeuralNetworks.cpp127 static_assert(ANEURALNETWORKS_ARGMAX == 39, "ANEURALNETWORKS_ARGMAX has changed");
401 static_assert(static_cast<int32_t>(OperationType::ARGMAX) == ANEURALNETWORKS_ARGMAX,
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp503 if ((mOpCode == ANEURALNETWORKS_ARGMIN || mOpCode == ANEURALNETWORKS_ARGMAX) && in testMutatingInputOperandCode()
1057 argMinMaxTest(ANEURALNETWORKS_ARGMAX, ANEURALNETWORKS_TENSOR_FLOAT16); in TEST()
1058 argMinMaxTest(ANEURALNETWORKS_ARGMAX, ANEURALNETWORKS_TENSOR_FLOAT32); in TEST()
1059 argMinMaxTest(ANEURALNETWORKS_ARGMAX, ANEURALNETWORKS_TENSOR_INT32); in TEST()
1060 argMinMaxTest(ANEURALNETWORKS_ARGMAX, ANEURALNETWORKS_TENSOR_QUANT8_ASYMM); in TEST()
1061 argMinMaxTest(ANEURALNETWORKS_ARGMAX, ANEURALNETWORKS_TENSOR_QUANT8_ASYMM_SIGNED); in TEST()
/frameworks/ml/nn/common/
DUtils.cpp1614 case ANEURALNETWORKS_ARGMAX: in validateOperation()