Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDABlas.cpp383 uint32_t b_alignment = _getAlignment(reinterpret_cast<uintptr_t>(b)); in bgemm_internal_cublaslt() local
1251 uint32_t b_alignment = _getAlignment(reinterpret_cast<uintptr_t>(mat2_ptr)); in gemm_and_bias() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConv_v8.cpp183 uint8_t b_alignment; member