at::native Namespace Reference

Classes
struct	ActivationDescriptor
union	Constant
struct	ConvolutionDescriptor
struct	CTCLossDescriptor
class	Descriptor
struct	DescriptorDeleter
class	DftiDescriptor
struct	DftiDescriptorDeleter
struct	DropoutDescriptor
class	FilterDescriptor
struct	RNNDescriptor
struct	SpatialTransformerDescriptor
class	TensorDescriptor

Functions
int	dataSize (cudnnDataType_t dataType)
template<typename T >
static void	fixSizeOneDimStride (int dim, const T *size, T *stride, bool nhwc)
std::ostream &	operator<< (std::ostream &out, const TensorDescriptor &d)
std::ostream &	operator<< (std::ostream &out, const FilterDescriptor &d)
TORCH_CUDA_CPP_API cudnnDataType_t	getCudnnDataTypeFromScalarType (const at::ScalarType dtype)
cudnnDataType_t	getCudnnDataType (const at::Tensor &tensor)
Tensor	contiguousIfZeroInStrides (const Tensor &t)
at::Tensor	clone_preserve_strides (const at::Tensor &self)
bool	cat_should_skip_tensor (const Tensor &t)
void	check_cat_shape_except_dim (const Tensor &first, const Tensor &second, int64_t dimension, int64_t index)
void	check_cat_no_zero_dim (const MaterializedITensorListRef &tensors)
int64_t	get_num_splits (const Tensor &self, int64_t split_size, int64_t dim)

Function Documentation

cat_should_skip_tensor()

bool at::native::cat_should_skip_tensor ( const Tensor &  t )

inline

check_cat_no_zero_dim()

void at::native::check_cat_no_zero_dim ( const MaterializedITensorListRef &  tensors )

inline

check_cat_shape_except_dim()

void at::native::check_cat_shape_except_dim ( const Tensor &  first, const Tensor &  second, int64_t  dimension, int64_t  index  )

inline

clone_preserve_strides()

at::Tensor at::native::clone_preserve_strides

(

const at::Tensor &

self

)

contiguousIfZeroInStrides()

Tensor at::native::contiguousIfZeroInStrides ( const Tensor &  t )

inline

dataSize()

int at::native::dataSize ( cudnnDataType_t  dataType )

inline

fixSizeOneDimStride()

template<typename T >

static void at::native::fixSizeOneDimStride ( int  dim, const T *  size, T *  stride, bool  nhwc  )

inlinestatic

get_num_splits()

int64_t at::native::get_num_splits ( const Tensor &  self, int64_t  split_size, int64_t  dim  )

inline

getCudnnDataType()

cudnnDataType_t at::native::getCudnnDataType

(

const at::Tensor &

tensor

)

getCudnnDataTypeFromScalarType()

TORCH_CUDA_CPP_API cudnnDataType_t at::native::getCudnnDataTypeFromScalarType

(

const at::ScalarType

dtype

)

operator<<() [1/2]

std::ostream & at::native::operator<<	(	std::ostream &	out,
		const FilterDescriptor &	d
	)

operator<<() [2/2]

std::ostream & at::native::operator<<	(	std::ostream &	out,
		const TensorDescriptor &	d
	)