/aosp_15_r20/external/ruy/ruy/ |
H A D | kernel_arm.h | 58 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 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>;
|
H A D | pack_x86.h | 66 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 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()
|
H A D | kernel_x86.h | 45 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 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>, \
|
H A 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;
|
H A 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,
|
H A D | pack_avx2_fma.cc | 57 PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kColMajor, 4, 8>, 61 PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
|
H A 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,
|
H A D | pack_avx512.cc | 63 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/ |
D | neon_tensor_utils.sdump | 8488 …"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 …]
|