Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/iterators/
Depilogue_predicated_tile_iterator.h526 int output_N = output_row / (convolution_P * convolution_Q); in downsample_load_with_byte_offset() local
601 int output_N = output_row / (convolution_P * convolution_Q); in upsample_load_with_byte_offset() local