opennn::CudnnRnnShapeSlot Struct Reference

#include <opennn/network/operators/cudnn_rnn.h>

Public Attributes
Index	batch = -1
Index	time = -1
Index	input_features = -1
int	stamp = 0
bool	training_ready = false
bool	time_major = true
Index	workspace_bytes = 0
Index	reserve_space_bytes = 0
CudnnDescriptor< DnnRNNDataDescriptor >	x_desc
CudnnDescriptor< DnnRNNDataDescriptor >	y_desc
CudnnDescriptor< DnnTensorDescriptor >	h_desc
CudnnDescriptor< DnnTensorDescriptor >	c_desc
Buffer	seq_host {Device::CPU}
Buffer	seq_dev {Device::CUDA}

Member Data Documentation

batch

Index opennn::CudnnRnnShapeSlot::batch = -1

c_desc

CudnnDescriptor<DnnTensorDescriptor> opennn::CudnnRnnShapeSlot::c_desc

h_desc

CudnnDescriptor<DnnTensorDescriptor> opennn::CudnnRnnShapeSlot::h_desc

input_features

Index opennn::CudnnRnnShapeSlot::input_features = -1

reserve_space_bytes

Index opennn::CudnnRnnShapeSlot::reserve_space_bytes = 0

seq_dev

Buffer opennn::CudnnRnnShapeSlot::seq_dev {Device::CUDA}

seq_host

Buffer opennn::CudnnRnnShapeSlot::seq_host {Device::CPU}

stamp

int opennn::CudnnRnnShapeSlot::stamp = 0

time

Index opennn::CudnnRnnShapeSlot::time = -1

time_major

bool opennn::CudnnRnnShapeSlot::time_major = true

training_ready

bool opennn::CudnnRnnShapeSlot::training_ready = false

workspace_bytes

Index opennn::CudnnRnnShapeSlot::workspace_bytes = 0

x_desc

CudnnDescriptor<DnnRNNDataDescriptor> opennn::CudnnRnnShapeSlot::x_desc

y_desc

CudnnDescriptor<DnnRNNDataDescriptor> opennn::CudnnRnnShapeSlot::y_desc