Searched refs:frsqrt (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Elementwise.cpp | 112 const std::function<float(float)> frsqrt = [](float x) { return 1.f / std::sqrt(x); }; in executeRsqrt() local
116 return compute<float, _Float16>(frsqrt, context->getInputBuffer<_Float16>(kInputTensor), in executeRsqrt()
120 return compute<float, float>(frsqrt, context->getInputBuffer<float>(kInputTensor), in executeRsqrt()
127 makeQuantized(frsqrt, inShape.scale, static_cast<uint8_t>(inShape.offset), in executeRsqrt()
137 makeQuantized(frsqrt, inShape.scale, static_cast<int8_t>(inShape.offset), in executeRsqrt()
|