Searched refs:seqLengthArray (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/ |
H A D | Descriptors.h | 128 …NNDataLayout_t layout, int maxSeqLength, int batchSize, int vectorSize, const int* seqLengthArray); 130 …NDataLayout_t layout, int maxSeqLength, int batchSize, int vectorSize, const int* seqLengthArray) { in set() argument 131 …scriptor(mut_desc(), dataType, layout, maxSeqLength, batchSize, vectorSize, seqLengthArray, NULL)); in set()
|
H A D | Descriptors.cpp | 32 …ut, const int maxSeqLength, const int batchSize, const int vectorSize, const int* seqLengthArray) { in set() argument 33 set(getDataType(t), layout, maxSeqLength, batchSize, vectorSize, seqLengthArray); in set()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
H A D | RNN.cpp | 319 std::vector<int> seqLengthArray(batch_size, 1); in rnn_descriptor_sequence() local 333 seqLengthArray[idx]++; in rnn_descriptor_sequence() 342 seqLengthArray.data()); in rnn_descriptor_sequence() 356 std::vector<int32_t> seqLengthArray(batch_size, seq_len); in rnn_descriptor() local 358 tensor, layout, seq_len, batch_size, vector_size, seqLengthArray.data()); in rnn_descriptor()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
H A D | cudnn_7_6.inc | 2317 const int seqLengthArray[], /* length of each sequence in the batch */ 2325 vectorSize, seqLengthArray, paddingFill); 2331 int *vectorSize, int arrayLengthRequested, int seqLengthArray[], 2339 vectorSize, arrayLengthRequested, seqLengthArray, 2669 size_t seqLengthArraySize, const int seqLengthArray[], void *paddingFill) { 2676 seqLengthArray, paddingFill); 2683 int seqLengthArray[], void *paddingFill) { 2690 seqLengthArraySize, seqLengthSizeRequested, seqLengthArray,
|
H A D | cudnn_8_0.inc | 1993 const int seqLengthArray[], /* length of each sequence in the batch */ 2001 vectorSize, seqLengthArray, paddingFill); 2007 int *vectorSize, int arrayLengthRequested, int seqLengthArray[], 2015 vectorSize, arrayLengthRequested, seqLengthArray, 2074 size_t seqLengthArraySize, const int seqLengthArray[], void *paddingFill) { 2081 seqLengthArray, paddingFill); 2088 int seqLengthArray[], void *paddingFill) { 2095 seqLengthArraySize, seqLengthSizeRequested, seqLengthArray,
|
H A D | cudnn_7_3.inc | 2397 const int seqLengthArray[], /* length of each sequence in the batch */ 2405 vectorSize, seqLengthArray, paddingFill); 2411 int *vectorSize, int arrayLengthRequested, int seqLengthArray[], 2419 vectorSize, arrayLengthRequested, seqLengthArray,
|
H A D | cudnn_7_4.inc | 2538 const int seqLengthArray[], /* length of each sequence in the batch */ 2546 vectorSize, seqLengthArray, paddingFill); 2552 int *vectorSize, int arrayLengthRequested, int seqLengthArray[], 2560 vectorSize, arrayLengthRequested, seqLengthArray,
|