Searched defs:seqLengthArray (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/cudnn/ |
| D | Descriptors.cpp | 32 …ut, const int maxSeqLength, const int batchSize, const int vectorSize, const int* seqLengthArray) { in set()
|
| D | Descriptors.h | 130 …NDataLayout_t layout, int maxSeqLength, int batchSize, int vectorSize, const int* seqLengthArray) { in set()
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | RNN.cpp | 319 std::vector<int> seqLengthArray(batch_size, 1); in rnn_descriptor_sequence() local
356 std::vector<int32_t> seqLengthArray(batch_size, seq_len); in rnn_descriptor() local
|