Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
Dint4mm.cu327 static constexpr int32_t kNTileSize = 16; member
329 static constexpr int32_t kNTileSize = 8; member
451 static constexpr int32_t kNTileSize = 16; member
453 static constexpr int32_t kNTileSize = 8; member
631 constexpr int32_t kNTileSize = 16; in __launch_bounds__() local
633 constexpr int32_t kNTileSize = 8; in __launch_bounds__() local
1010 constexpr int32_t kNTileSize = 16; in matrix_to_m16n8k16_Bint4_layout() local
1012 constexpr int32_t kNTileSize = 8; in matrix_to_m16n8k16_Bint4_layout() local
1115 constexpr int32_t kNTileSize = 16; in _weight_int4pack_mm_cuda() local
1117 constexpr int32_t kNTileSize = 8; in _weight_int4pack_mm_cuda() local
[all …]
/external/pytorch/aten/src/ATen/native/
DLinearAlgebra.cpp3448 constexpr int64_t kNTileSize = 8; in _convert_weight_to_int4pack_cpu() local
3473 constexpr int64_t kNTileSize = 8; in _weight_int4pack_mm_cpu() local