Home
last modified time | relevance | path

Searched refs:seqLengthArray (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/
H A DDescriptors.h128 …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 DDescriptors.cpp32 …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 DRNN.cpp319 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 Dcudnn_7_6.inc2317 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 Dcudnn_8_0.inc1993 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 Dcudnn_7_3.inc2397 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 Dcudnn_7_4.inc2538 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,