Home
last modified time | relevance | path

Searched refs:FixedKernelLayout (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/ruy/ruy/
H A Dkernel_arm.h58 using LhsLayout = FixedKernelLayout<Order::kColMajor, 16, 4>;
59 using RhsLayout = FixedKernelLayout<Order::kColMajor, 16, 4>;
86 using LhsLayout = FixedKernelLayout<Order::kColMajor, 16, 4>;
87 using RhsLayout = FixedKernelLayout<Order::kColMajor, 16, 2>;
112 using LhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
113 using RhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
147 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
148 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
173 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
174 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 4>;
[all …]
H A Dkernel.h146 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>;
H A Dpack_x86.h66 struct PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kColMajor, 4, 8>,
71 using Layout = FixedKernelLayout<Order::kColMajor, 4, 8>;
113 struct PackImpl<Path::kAvx, FixedKernelLayout<Order::kColMajor, 4, 8>, Scalar,
118 using Layout = FixedKernelLayout<Order::kColMajor, 4, 8>;
159 struct PackImpl<Path::kAvx, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
161 using Layout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
193 struct PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kRowMajor, 1, 8>,
195 using Layout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
231 struct PackImpl<Path::kAvx512, FixedKernelLayout<Order::kColMajor, 4, 16>,
236 using Layout = FixedKernelLayout<Order::kColMajor, 4, 16>;
[all …]
H A Dpack.h105 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()
H A Dkernel_x86.h45 using LhsLayout = FixedKernelLayout<Order::kColMajor, 4, 16>;
46 using RhsLayout = FixedKernelLayout<Order::kColMajor, 4, 16>;
68 using LhsLayout = FixedKernelLayout<Order::kColMajor, 4, 16>;
69 using RhsLayout = FixedKernelLayout<Order::kColMajor, 4, 16>;
93 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 16>;
94 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 16>;
119 using LhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
120 using RhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
142 using LhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
143 using RhsLayout = FixedKernelLayout<Order::kColMajor, 4, 8>;
[all …]
H A Dpack_common.h60 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>, \
H A Dmat.h465 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;
H A Dpack_arm.h143 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,
H A Dpack_avx2_fma.cc57 PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kColMajor, 4, 8>,
61 PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
H A Dpack_avx.cc56 PackImpl<Path::kAvx, FixedKernelLayout<Order::kColMajor, 4, 8>, std::int8_t,
60 PackImpl<Path::kAvx, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
H A Dpack_avx512.cc63 PackImpl<Path::kAvx512, FixedKernelLayout<Order::kColMajor, 4, 16>,
66 PackImpl<Path::kAvx512, FixedKernelLayout<Order::kColMajor, 4, 16>,
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/kernels/libtflite_kernel_utils/android_arm64_armv8-2a_cortex-a55_static_apex30_sabi/obj/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.sdump8488 …"function_name" : "ruy::PackImpl<ruy::Path::kNeon, ruy::FixedKernelLayout<ruy::Order::kColMajor, 1…
8512 …"function_name" : "ruy::PackImpl<ruy::Path::kNeon, ruy::FixedKernelLayout<ruy::Order::kColMajor, 1…
8536 …"function_name" : "ruy::PackImpl<ruy::Path::kNeon, ruy::FixedKernelLayout<ruy::Order::kRowMajor, 1…
8560 …"function_name" : "ruy::PackImpl<ruy::Path::kStandardCpp, ruy::FixedKernelLayout<ruy::Order::kColM…
8584 …"function_name" : "ruy::PackImpl<ruy::Path::kStandardCpp, ruy::FixedKernelLayout<ruy::Order::kColM…
8608 …"function_name" : "ruy::PackImpl<ruy::Path::kNeonDotprod, ruy::FixedKernelLayout<ruy::Order::kColM…
8632 …"function_name" : "ruy::PackImpl<ruy::Path::kNeonDotprod, ruy::FixedKernelLayout<ruy::Order::kColM…
68823 "name" : "ruy::FixedKernelLayout::kCols",
68829 "name" : "ruy::FixedKernelLayout::kRows",
68835 "name" : "ruy::FixedKernelLayout::kOrder",
[all …]