Home
last modified time | relevance | path

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

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/gemm/f32/
H A Dgemm_utils_f32.hpp28 struct gemm_traits {}; struct
31 struct gemm_traits<double, isTransA, isTransB> { struct
32 static constexpr dim_t m = 8;
33 static constexpr dim_t n = 6;
34 static constexpr dim_t BM = 4032;
35 static constexpr dim_t BN = isTransA ? 96 : 192;
40 struct gemm_traits<float, isTransA, isTransB> { struct
41 static constexpr dim_t m = 16;
42 static constexpr dim_t n = 6;
43 static constexpr dim_t BM = 4032;
[all …]
/dports/math/onednn/oneDNN-2.5.1/src/cpu/gemm/f32/
H A Dgemm_utils_f32.hpp28 struct gemm_traits {}; struct
31 struct gemm_traits<double, isTransA, isTransB> { struct
32 static constexpr dim_t m = 8;
33 static constexpr dim_t n = 6;
34 static constexpr dim_t BM = 4032;
35 static constexpr dim_t BN = isTransA ? 96 : 192;
40 struct gemm_traits<float, isTransA, isTransB> { struct
41 static constexpr dim_t m = 16;
42 static constexpr dim_t n = 6;
43 static constexpr dim_t BM = 4032;
[all …]