| /external/ruy/ruy/ |
| D | kernel_arm.h | 57 using LhsLayout = FixedKernelLayout<Order::kColMajor, 16, 4>;
58 using RhsLayout = FixedKernelLayout<Order::kColMajor, 16, 4>;
85 using LhsLayout = FixedKernelLayout<Order::kColMajor, 16, 4>;
86 using RhsLayout = FixedKernelLayout<Order::kColMajor, 16, 2>;
110 using LhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
111 using RhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
142 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
143 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
166 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
167 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 4>;
[all …]
|
| D | pack.h | 105 template <Path ThePath, typename FixedKernelLayout, typename Scalar,
113 RUY_DCHECK_EQ((end_col - start_col) % FixedKernelLayout::kCols, 0); in Run()
135 template <Path ThePath, typename FixedKernelLayout, typename Scalar,
145 PackImpl<ThePath, FixedKernelLayout, Scalar, PackedScalar, SumsType, in RunPack()
148 PackImpl<ThePath, FixedKernelLayout, Scalar, PackedScalar, SumsType, in RunPack()
|
| D | pack_common.h | 60 template <Path ThePath, typename FixedKernelLayout, typename Scalar,
65 template <typename FixedKernelLayout, typename Scalar, \
67 struct PackImpl<CHILD, FixedKernelLayout, Scalar, PackedScalar, SumsType, \
68 SrcOrder> : PackImpl<PARENT, FixedKernelLayout, Scalar, \
137 struct PackImpl<ThePath, FixedKernelLayout<Order::kRowMajor, 1, KernelCols>, \
|
| D | kernel.h | 146 using Lhs = FixedKernelLayout<Order::kColMajor, 1, 1>;
147 using Rhs = FixedKernelLayout<Order::kColMajor, 1, 1>;
153 using Lhs = FixedKernelLayout<Order::kRowMajor, 4, 4>;
154 using Rhs = FixedKernelLayout<Order::kRowMajor, 4, 4>;
160 using Lhs = FixedKernelLayout<Order::kColMajor, 1, 4>;
161 using Rhs = FixedKernelLayout<Order::kColMajor, 1, 8>;
167 using Lhs = FixedKernelLayout<Order::kColMajor, 2, 16>;
168 using Rhs = FixedKernelLayout<Order::kRowMajor, 2, 8>;
|
| D | pack_x86.h | 65 struct PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kColMajor, 4, 8>,
70 using Layout = FixedKernelLayout<Order::kColMajor, 4, 8>;
112 struct PackImpl<Path::kAvx, FixedKernelLayout<Order::kColMajor, 4, 8>, Scalar,
117 using Layout = FixedKernelLayout<Order::kColMajor, 4, 8>;
158 struct PackImpl<Path::kAvx, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
160 using Layout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
192 struct PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kRowMajor, 1, 8>,
194 using Layout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
230 struct PackImpl<Path::kAvx512, FixedKernelLayout<Order::kColMajor, 4, 16>,
235 using Layout = FixedKernelLayout<Order::kColMajor, 4, 16>;
[all …]
|
| D | mat.h | 465 struct FixedKernelLayout { struct
471 template <typename FixedKernelLayout> argument
474 ret.order = FixedKernelLayout::kOrder; in ToKernelLayout()
475 ret.rows = FixedKernelLayout::kRows; in ToKernelLayout()
476 ret.cols = FixedKernelLayout::kCols; in ToKernelLayout()
485 constexpr int FixedKernelLayout<tOrder, tRows, tCols>::kCols;
487 constexpr int FixedKernelLayout<tOrder, tRows, tCols>::kRows;
|
| D | kernel_x86.h | 45 using LhsLayout = FixedKernelLayout<Order::kColMajor, 4, 16>;
46 using RhsLayout = FixedKernelLayout<Order::kColMajor, 4, 16>;
70 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 16>;
71 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 16>;
96 using LhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
97 using RhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
121 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
122 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
146 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
147 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
[all …]
|
| D | pack_arm.h | 143 struct PackImpl<Path::kNeon, FixedKernelLayout<Order::kColMajor, 16, 4>, Scalar,
227 struct PackImpl<Path::kNeon, FixedKernelLayout<Order::kColMajor, 16, 2>, Scalar,
275 struct PackImpl<Path::kNeonDotprod, FixedKernelLayout<Order::kColMajor, 4, 8>,
363 struct PackImpl<Path::kNeon, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
439 struct PackImpl<Path::kNeon, FixedKernelLayout<Order::kRowMajor, 1, 4>, float,
498 struct PackImpl<Path::kNeonDotprod, FixedKernelLayout<Order::kColMajor, 4, 8>,
579 FixedKernelLayout<Order::kColMajor, 16, KernelCols>, Scalar,
|
| D | pack_avx2_fma.cc | 57 PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kColMajor, 4, 8>,
61 PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
|
| D | pack_avx.cc | 56 PackImpl<Path::kAvx, FixedKernelLayout<Order::kColMajor, 4, 8>, std::int8_t,
60 PackImpl<Path::kAvx, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
|
| D | pack_avx512.cc | 57 PackImpl<Path::kAvx512, FixedKernelLayout<Order::kColMajor, 4, 16>,
|