Searched refs:kAvx2Fma (Results 1 – 9 of 9) sorted by relevance
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/third_party/ruy/ruy/ |
| A D | ctx_test.cc | 36 ctx.SetRuntimeEnabledPaths(Path::kAvx2Fma); in TEST()
38 EXPECT_EQ(ruy_paths, Path::kStandardCpp | Path::kAvx2Fma); in TEST()
|
| 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,
379 t0 = MaskLoadu<Path::kAvx2Fma>(available_src_rows, zero_point, src_ptr0);
380 t4 = MaskLoadu<Path::kAvx2Fma>(available_src_rows, zero_point, src_ptr4);
381 t1 = MaskLoadu<Path::kAvx2Fma>(available_src_rows, zero_point, src_ptr1);
382 t5 = MaskLoadu<Path::kAvx2Fma>(available_src_rows, zero_point, src_ptr5);
383 t2 = MaskLoadu<Path::kAvx2Fma>(available_src_rows, zero_point, src_ptr2);
384 t6 = MaskLoadu<Path::kAvx2Fma>(available_src_rows, zero_point, src_ptr6);
385 t3 = MaskLoadu<Path::kAvx2Fma>(available_src_rows, zero_point, src_ptr3);
538 inline __m256i CompareGreaterThan<Path::kAvx2Fma>(const __m256i& a,
[all …]
|
| A D | path.h | 85 kAvx2Fma = 0x20, enumerator
151 constexpr Path kDefaultArchPaths = Path::kAvx | Path::kAvx2Fma | Path::kAvx512;
|
| A D | pack_x86.h | 37 RUY_INHERIT_PACK(Path::kAvx, Path::kAvx2Fma)
38 RUY_INHERIT_PACK(Path::kAvx2Fma, Path::kAvx512)
40 RUY_USE_MEMCPY_ROWMAJOR_FLOAT_PACK(Path::kAvx2Fma, 8)
49 struct PackedTypeImpl<Path::kAvx2Fma, std::uint8_t> {
65 struct PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kColMajor, 4, 8>,
192 struct PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kRowMajor, 1, 8>,
315 struct PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kColMajor, 4, 8>,
|
| A D | kernel_x86.h | 34 RUY_INHERIT_KERNEL(Path::kStandardCpp, Path::kAvx2Fma)
36 RUY_INHERIT_KERNEL(Path::kAvx2Fma, Path::kAvx512)
92 struct Kernel<Path::kAvx2Fma, std::int8_t, std::int8_t, std::int32_t,
94 static constexpr Path kPath = Path::kAvx2Fma;
118 struct Kernel<Path::kAvx2Fma, float, float, float, float> {
119 static constexpr Path kPath = Path::kAvx2Fma;
169 static constexpr Path kPath = Path::kAvx2Fma;
|
| A D | kernel_avx2_fma.cc | 64 inline __m256i mm256_shuffle_epi8<Path::kAvx2Fma>(const __m256i& a,
72 inline __m256 MulAdd<Path::kAvx2Fma>(const __m256& a, const __m256& b,
78 inline __m128i mm256_extracti128_si256<Path::kAvx2Fma>(const __m256i& a,
725 Kernel8bitAvx2Impl<Path::kAvx2Fma>(params);
996 Kernel8bitAvx2SingleColImpl<Path::kAvx2Fma>(params);
1001 KernelFloatAvxCommon<Path::kAvx2Fma>(params);
1006 KernelFloatAvxCommonSingleCol<Path::kAvx2Fma>(params);
|
| A D | ctx.cc | 124 maybe_add(Path::kAvx2Fma, in DetectRuntimeSupportedPaths()
|
| A D | trace.h | 525 RUY_HANDLE_PATH(kAvx2Fma) in str()
|
| A D | test.h | 108 RUY_PATHNAME_CASE(kAvx2Fma) in PathName()
|
Completed in 20 milliseconds