1 // 2 // Copyright © 2019 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <iostream> 9 10 namespace armnn 11 { 12 template<typename T> 13 struct rsqrt : public std::unary_function<T, T> 14 { 15 T operator ()armnn::rsqrt16 operator () (const T& inputData) const 17 { 18 return 1 / std::sqrt(inputData); 19 } 20 }; 21 22 } //namespace armnn 23