Searched defs:ld_src (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/cpu/vec/vec512/ |
H A D | vec512_float.h | 585 inline void transpose_mxn_16x16(const float* src, int64_t ld_src, float* dst, int64_t ld_dst, int M… in transpose_mxn_16x16() 702 inline void transpose_mxn(const float* src, int64_t ld_src, float* dst, int64_t ld_dst) { in transpose_mxn()
|
H A D | vec512_bfloat16.h | 1388 inline void transpose_mxn(const BFloat16* src, int64_t ld_src, BFloat16* dst, int64_t ld_dst) { in transpose_mxn() 1430 inline void transpose_mxn(const Half* src, int64_t ld_src, Half* dst, int64_t ld_dst) { in transpose_mxn()
|