Home
last modified time | relevance | path

Searched defs:output_PQ (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/iterators/
Depilogue_predicated_tile_iterator.h527 int output_PQ = output_row % (convolution_P * convolution_Q); in downsample_load_with_byte_offset() local
602 int output_PQ = output_row % (convolution_P * convolution_Q); in upsample_load_with_byte_offset() local