pow.h

Go to the documentation of this file.

#pragma once

#include <cmath>
#include "tensor/tensor.h"

namespace magmadnn {
namespace math {


template <typename T>
void pow(Tensor<T> *x, int power, Tensor<T> *out);

#if defined(_HAS_CUDA_)
template <typename T>
void pow_device(Tensor<T> *x, int power, Tensor<T> *out);
#endif


}
}