Line data Source code
1 : #pragma once 2 : 3 : #include "functions/Square.hpp" 4 : 5 : #include <thrust/transform.h> 6 : 7 : namespace elsa 8 : { 9 : /// @brief Compute the square for each element of the input range 10 : /// @ingroup transforms 11 : template <class InputIter, class OutIter> 12 : void square(InputIter first, InputIter last, OutIter out) 13 159 : { 14 159 : thrust::transform(first, last, out, elsa::fn::square); 15 159 : } 16 : } // namespace elsa