Searched defs:RNNDescriptorParams (Results 1 – 2 of 2) sorted by relevance
72 struct RNNDescriptorParams { struct73 int64_t hidden_size;74 int64_t num_layers;75 miopenRNNDirectionMode_t direction;76 miopenRNNMode_t rnn_mode;77 miopenDataType_t datatype;78 miopenRNNAlgo_t algo = miopenRNNdefault;79 miopenRNNInputMode_t input_mode = miopenRNNlinear;80 miopenRNNBiasMode_t bias_mode = miopenRNNNoBias;82 int64_t num_directions() const { in num_directions()[all …]
148 struct RNNDescriptorParams { struct150 int64_t input_size;151 bool packed;153 int64_t hidden_size;154 int64_t proj_size;155 int64_t num_layers;156 cudnnDirectionMode_t bidirectional;157 cudnnRNNMode_t mode;158 cudnnDataType_t datatype;159 cudnnDataType_t input_datatype;[all …]