Descriptors.h File Reference

#include <string>
#include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/Exceptions.h>
#include <ATen/cudnn/cudnn-wrapper.h>
#include <ATen/cudnn/Utils.h>
#include <ATen/core/Tensor.h>
#include <ATen/TensorUtils.h>
#include <ATen/cuda/ATenCUDAGeneral.h>
#include <cuda.h>
#include <ATen/Functions.h>

Go to the source code of this file.

Classes
struct	at::native::DescriptorDeleter< T, dtor >
class	at::native::Descriptor< T, ctor, dtor >
class	at::native::TensorDescriptor
class	at::native::FilterDescriptor
struct	at::native::ConvolutionDescriptor
struct	at::native::SpatialTransformerDescriptor
struct	at::native::DropoutDescriptor
struct	at::native::RNNDescriptor
struct	at::native::CTCLossDescriptor
struct	at::native::ActivationDescriptor
union	at::native::Constant

Namespaces
namespace	at
namespace	at::native

Functions
int	at::native::dataSize (cudnnDataType_t dataType)
template<typename T >
static void	at::native::fixSizeOneDimStride (int dim, const T *size, T *stride, bool nhwc)
std::ostream &	at::native::operator<< (std::ostream &out, const TensorDescriptor &d)
std::ostream &	at::native::operator<< (std::ostream &out, const FilterDescriptor &d)