etl::impl::cudnn::cudnn_wrapper< T > Struct Template Reference

Wrapper for CUDNN tensor. More...

#include <cudnn.hpp>

Collaboration diagram for etl::impl::cudnn::cudnn_wrapper< T >:

Public Member Functions
	cudnn_wrapper (T tensor)
	Create a new cudnn_wrapper for the given tensor.
	cudnn_wrapper (cudnn_wrapper &&rhs)
	Move construct a cudnn_wrapper. More...
	cudnn_wrapper (const cudnn_wrapper &rhs)=delete
cudnn_wrapper &	operator= (const cudnn_wrapper &rhs)=delete
cudnn_wrapper &	operator= (cudnn_wrapper &&rhs)=delete
T	operator* ()
	Convert back to the CUDNN tensor type. More...
	~cudnn_wrapper ()
	Delete the cudnn_wrapper, releasing the tensor.
template<>
	~cudnn_wrapper ()
template<>
	~cudnn_wrapper ()
template<>
	~cudnn_wrapper ()

Public Attributes
T	tensor
	The CUDNN tensor.

Detailed Description

template<typename T>
struct etl::impl::cudnn::cudnn_wrapper< T >

Wrapper for CUDNN tensor.

Needs to be specialized for the destructor.

Constructor & Destructor Documentation

cudnn_wrapper()

template<typename T >

etl::impl::cudnn::cudnn_wrapper< T >::cudnn_wrapper ( cudnn_wrapper< T > &&  rhs )

inline

Move construct a cudnn_wrapper.

This will remove the tensor from the RHS.

Parameters

rhs	The right hand side cudnn_wrapper

~cudnn_wrapper() [1/3]

template<>

etl::impl::cudnn::cudnn_wrapper< cudnnTensorDescriptor_t >::~cudnn_wrapper ( )

inline

~cudnn_wrapper() [2/3]

template<>

etl::impl::cudnn::cudnn_wrapper< cudnnFilterDescriptor_t >::~cudnn_wrapper ( )

inline

~cudnn_wrapper() [3/3]

template<>

etl::impl::cudnn::cudnn_wrapper< cudnnPoolingDescriptor_t >::~cudnn_wrapper ( )

inline

Member Function Documentation

operator*()

template<typename T >

T etl::impl::cudnn::cudnn_wrapper< T >::operator* ( )

inline

Convert back to the CUDNN tensor type.

Returns

The CUDNN tensor.

---

The documentation for this struct was generated from the following file:

• include/etl/impl/cudnn/cudnn.hpp