// | |
// 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 |