Searched refs:IntermediateType (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Elementwise.cpp | 35 template <typename IntermediateType, typename T>
36 inline bool compute(const std::function<IntermediateType(IntermediateType)>& func, const T* input, in compute()
40 output[i] = static_cast<T>(func(static_cast<IntermediateType>(input[i]))); in compute()
45 template <typename IntermediateType, typename T>
46 inline bool compute(IntermediateType func(IntermediateType), const T* input, const Shape& shape, in compute() argument
48 return compute(std::function<IntermediateType(IntermediateType)>(func), input, shape, output); in compute()
51 template <typename IntermediateType, typename T>
52 auto makeQuantized(const std::function<IntermediateType(IntermediateType)>& func, float inScale, in makeQuantized()
58 IntermediateType dequantizedVal = in makeQuantized()
61 IntermediateType res = func(dequantizedVal); in makeQuantized()
[all …]
|