Searched refs:ANEURALNETWORKS_SOFTMAX (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/include/ |
| D | NeuralNetworksTypes.h | 1920 ANEURALNETWORKS_SOFTMAX = 25, enumerator
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | NeuralNetworks.cpp | 113 static_assert(ANEURALNETWORKS_SOFTMAX == 25, "ANEURALNETWORKS_SOFTMAX has changed");
339 static_assert(static_cast<int32_t>(OperationType::SOFTMAX) == ANEURALNETWORKS_SOFTMAX,
|
| /packages/modules/NeuralNetworks/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()
|