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 14 { 15 typedef T result_type; 16 typedef T argument_type; 17 18 T operator ()armnn::rsqrt19 operator () (const T& inputData) const 20 { 21 return 1 / std::sqrt(inputData); 22 } 23 }; 24 25 } //namespace armnn 26