tanh_internal.h

Go to the documentation of this file.

#pragma once
#include <math.h>
#include "tensor/tensor.h"


namespace magmadnn {
namespace internal {

template <typename T>
void tanh_full(Tensor<T> *x, Tensor<T> *out);

#if defined(_HAS_CUDA_)

template <typename T>
void tanh_full_device(Tensor<T> *x, Tensor<T> *out); 
#endif

}   // namespace internal
}   // namespace magmadnn