TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
pytorch_quantization.nn.modules.quant_rnn Namespace Reference

Classes

class  GRU
 
class  GRUCell
 
class  QuantLSTM
 
class  QuantLSTMCell
 
class  QuantRNN
 
class  QuantRNNBase
 
class  QuantRNNCell
 
class  QuantRNNCellBase
 

Variables

list __all__ = ["QuantLSTM", "QuantLSTMCell", "LSTM", "LSTMCell"]
 
 LSTM = QuantLSTM
 
 LSTMCell = QuantLSTMCell
 

Variable Documentation

◆ __all__

list pytorch_quantization.nn.modules.quant_rnn.__all__ = ["QuantLSTM", "QuantLSTMCell", "LSTM", "LSTMCell"]
private

◆ LSTM

pytorch_quantization.nn.modules.quant_rnn.LSTM = QuantLSTM

◆ LSTMCell

pytorch_quantization.nn.modules.quant_rnn.LSTMCell = QuantLSTMCell