Searched refs:elementwise (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Elementwise.cpp | 28 namespace elementwise { namespace
175 NN_REGISTER_OPERATION(ABS, "ABS", elementwise::validateAbs, elementwise::prepare,
176 elementwise::executeAbs);
177 NN_REGISTER_OPERATION(EXP, "EXP", elementwise::validate, elementwise::prepare,
178 elementwise::executeExp);
179 NN_REGISTER_OPERATION(FLOOR, "FLOOR", elementwise::validateFloor, elementwise::prepareFloor,
180 elementwise::executeFloor);
181 NN_REGISTER_OPERATION(LOG, "LOG", elementwise::validate, elementwise::prepare,
182 elementwise::executeLog);
183 NN_REGISTER_OPERATION(RSQRT, "RSQRT", elementwise::validate, elementwise::prepare,
[all …]
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 3836 * For input tensors x and y, computes x == y elementwise.
4064 * For input tensors x and y, computes x > y elementwise.
4091 * For input tensors x and y, computes x >= y elementwise.
4403 * For input tensors x and y, computes x < y elementwise.
4431 * For input tensors x and y, computes x <= y elementwise.
4657 * For input tensors x and y, computes x != y elementwise.
4745 * base^exponent elementwise.
5337 * elementwise from both input tensors:
|