Searched refs:ANEURALNETWORKS_SOFTMAX (Results 1 – 6 of 6) sorted by relevance
/frameworks/ml/nn/runtime/test/generated/models/ |
D | softmax_float_2.model.cpp | 12 model->addOperation(ANEURALNETWORKS_SOFTMAX, {input, beta}, {output}); in CreateModel()
|
D | softmax_float_1.model.cpp | 12 model->addOperation(ANEURALNETWORKS_SOFTMAX, {input, beta}, {output}); in CreateModel()
|
D | softmax_quant8_1.model.cpp | 13 model->addOperation(ANEURALNETWORKS_SOFTMAX, {input, beta}, {output}); in CreateModel()
|
D | softmax_quant8_2.model.cpp | 13 model->addOperation(ANEURALNETWORKS_SOFTMAX, {input, beta}, {output}); in CreateModel()
|
/frameworks/ml/nn/runtime/include/ |
D | NeuralNetworks.h | 1045 ANEURALNETWORKS_SOFTMAX = 25, enumerator
|
/frameworks/ml/nn/runtime/ |
D | NeuralNetworks.cpp | 88 static_assert(ANEURALNETWORKS_SOFTMAX == 25, "ANEURALNETWORKS_SOFTMAX has changed"); 197 static_assert(static_cast<int32_t>(OperationType::SOFTMAX) == ANEURALNETWORKS_SOFTMAX,
|