crossentropy_internal.h

Go to the documentation of this file.

#pragma once

#include "tensor/tensor.h"

#if defined(_HAS_CUDA_)
#include <cuda.h>
#endif

namespace magmadnn {
namespace internal {

template <typename T>
void crossentropy_full(Tensor<T> *x, Tensor<T> *y, Tensor<T> *softmax, Tensor<T> *out);


#if defined(_HAS_CUDA_)
template <typename T>
void crossentropy_full_device(Tensor<T> *x, Tensor<T> *y, Tensor<T> *softmax, Tensor<T> *out);
#endif


}   // namespace internal
}   // namespace magmadnn