Searched refs:ANEURALNETWORKS_LOG_SOFTMAX (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | log_softmax.model.cpp | 17 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input0, param, param1}, {output0}); in CreateModel()
44 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input0, param, param1}, {output0}); in CreateModel_relaxed()
73 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input0, param, param1}, {output0}); in CreateModel_float16()
101 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input0, param, param1}, {output0}); in CreateModel_dynamic_output_shape()
129 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input0, param, param1}, {output0}); in CreateModel_dynamic_output_shape_relaxed()
159 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input0, param, param1}, {output0}); in CreateModel_dynamic_output_shape_float16()
186 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input01, param2, param3}, {output01}); in CreateModel_2()
213 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input01, param2, param3}, {output01}); in CreateModel_relaxed_2()
242 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input01, param2, param3}, {output01}); in CreateModel_float16_2()
270 model->addOperation(ANEURALNETWORKS_LOG_SOFTMAX, {input01, param2, param3}, {output01}); in CreateModel_dynamic_output_shape_2()
[all …]
|
| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Normalization.cpp | 204 .opType = ANEURALNETWORKS_LOG_SOFTMAX, in DEFINE_OPERATION_SIGNATURE()
|
| /frameworks/ml/nn/runtime/include/ |
| D | NeuralNetworks.h | 3627 ANEURALNETWORKS_LOG_SOFTMAX = 64, enumerator
|
| /frameworks/ml/nn/runtime/ |
| D | NeuralNetworks.cpp | 152 static_assert(ANEURALNETWORKS_LOG_SOFTMAX == 64, "ANEURALNETWORKS_LOG_SOFTMAX has changed");
435 static_assert(static_cast<int32_t>(OperationType::LOG_SOFTMAX) == ANEURALNETWORKS_LOG_SOFTMAX,
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 885 OperationTestBase test(ANEURALNETWORKS_LOG_SOFTMAX, {input, beta, axis}, {output}); in logSoftmaxOpTest()
|