Searched refs:ANEURALNETWORKS_SOFTMAX (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/runtime/include/ |
| D | NeuralNetworks.h | 1915 ANEURALNETWORKS_SOFTMAX = 25, enumerator
|
| /frameworks/ml/nn/runtime/ |
| D | NeuralNetworks.cpp | 109 static_assert(ANEURALNETWORKS_SOFTMAX == 25, "ANEURALNETWORKS_SOFTMAX has changed");
335 static_assert(static_cast<int32_t>(OperationType::SOFTMAX) == ANEURALNETWORKS_SOFTMAX,
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1802 OperationTestBase softmaxTest(ANEURALNETWORKS_SOFTMAX, {input, beta}, {output}, in softmaxOpTest()
1807 OperationTestBase softmaxAxisTest(ANEURALNETWORKS_SOFTMAX, {input, beta, axis}, {output}, in softmaxOpTest()
|
| D | TestValidateModel.cpp | 2900 ASSERT_EQ(ANeuralNetworksModel_addOperation(model, ANEURALNETWORKS_SOFTMAX, 2, input_84, 1, in TEST_F()
|