Power.hpp

Go to the documentation of this file.

//
// Copyright © 2023 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
 
#pragma once
 
#include <iostream>
 
namespace armnn
{
 
template<typename T>
struct power
{
    typedef T result_type;
    typedef T first_argument_type;
 
    T
    operator()(const T& input1, const T& input2) const
    {
        T power = armnn::numeric_cast<T>(std::pow(static_cast<float>(input1), static_cast<float>(input2)));
        return power;
    }
};
 
} //namespace armnn