Searched defs:RNNDescriptor (Results 1 – 2 of 2) sorted by relevance
122 struct RNNDescriptor struct129 …MIOPEN_CHECK(miopenSetRNNDescriptor(mut_desc(), hidden_size, num_layers, input_mode, direction, rn… in set() argument
266 struct TORCH_CUDA_CPP_API RNNDescriptor : public Descriptor< struct281 AT_CUDNN_CHECK(cudnnSetRNNDescriptor_v6( in set() argument