Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/nested/cuda/
DNestedTensorTransformerFunctions.cu50 const int grainsize = GRID_DIM_Y * BLOCK_DIM; in remove_padding_transform0213_2() local
91 const int grainsize = GRID_DIM_Y * BLOCK_DIM; in remove_padding_2() local
124 const int grainsize = GRID_DIM_Y * BLOCK_DIM; in remove_padding() local
260 const int grainsize = GRID_DIM_Y * BLOCK_DIM; in add_padding_1() local
299 const int grainsize = GRID_DIM_Y * BLOCK_DIM; in add_padding_2() local
344 const int grainsize = GRID_DIM_Y * BLOCK_DIM; in add_padding_3() local