Searched defs:RNNParams (Results 1 – 3 of 3) sorted by relevance
85 struct RNNParams { struct86 ideep::rnn_kind mode;87 int64_t seq_length;88 int64_t mini_batch;89 int64_t input_size;90 int64_t hidden_size;91 int64_t num_directions;92 int64_t num_layers;93 bool batch_first;94 bool train;[all …]
200 struct RNNParams { struct201 RNNDescriptorParams rnn;202 TensorDescriptorListParams tensors;
490 struct RNNParams { struct491 DropoutDescriptorParams dropout;492 RNNDescriptorParams rnn;493 TensorDescriptorListParams tensors;