Searched refs:elementwise (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Elementwise.cpp | 32 namespace elementwise { namespace
186 NN_REGISTER_OPERATION_DEFAULT_VALIDATION(ABS, elementwise::prepare, elementwise::executeAbs);
187 NN_REGISTER_OPERATION_DEFAULT_VALIDATION(EXP, elementwise::prepare, elementwise::executeExp);
188 NN_REGISTER_OPERATION_DEFAULT_VALIDATION(FLOOR, elementwise::prepareFloor,
189 elementwise::executeFloor);
190 NN_REGISTER_OPERATION_DEFAULT_VALIDATION(LOG, elementwise::prepare, elementwise::executeLog);
191 NN_REGISTER_OPERATION_DEFAULT_VALIDATION(RSQRT, elementwise::prepare, elementwise::executeRsqrt);
192 NN_REGISTER_OPERATION_DEFAULT_VALIDATION(SIN, elementwise::prepare, elementwise::executeSin);
193 NN_REGISTER_OPERATION_DEFAULT_VALIDATION(SQRT, elementwise::prepare, elementwise::executeSqrt);
|
| /packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | Elementwise.cpp | 22 namespace elementwise { namespace
86 NN_DEFINE_VALIDATION_FUNCTION(FLOOR, elementwise::validateFloor);
87 NN_DEFINE_VALIDATION_FUNCTION(ABS, elementwise::validateAbs);
88 NN_DEFINE_VALIDATION_FUNCTION(EXP, elementwise::validate);
89 NN_DEFINE_VALIDATION_FUNCTION(LOG, elementwise::validate);
90 NN_DEFINE_VALIDATION_FUNCTION(RSQRT, elementwise::validateRsqrt);
91 NN_DEFINE_VALIDATION_FUNCTION(SIN, elementwise::validate);
92 NN_DEFINE_VALIDATION_FUNCTION(SQRT, elementwise::validate);
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | Elementwise.h | 22 namespace android::nn::elementwise {
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3999 * For input tensors x and y, computes x == y elementwise.
4227 * For input tensors x and y, computes x > y elementwise.
4254 * For input tensors x and y, computes x >= y elementwise.
4567 * For input tensors x and y, computes x < y elementwise.
4595 * For input tensors x and y, computes x <= y elementwise.
4821 * For input tensors x and y, computes x != y elementwise.
4909 * base^exponent elementwise.
|