Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
Dint4mm.cu324 struct ALayout_RM { struct
325 static constexpr int32_t kMTileSize = 16;
327 static constexpr int32_t kNTileSize = 16;
329 static constexpr int32_t kNTileSize = 8;
331 static constexpr int32_t kKTileSize = 16;
334 static __device__ void load( in load()
391 static __device__ void store( in store()