Searched refs:KernelCols (Results 1 – 2 of 2) sorted by relevance
/external/ruy/ruy/ |
D | pack_common.h | 100 template <Path ThePath, int KernelCols> 106 RUY_DCHECK_EQ(start_col % KernelCols, 0); in Run() 117 KernelCols * block_row; in Run() 120 for (; col <= src_cols - KernelCols; col += KernelCols) { in Run() 121 memcpy(packed_ptr, src_ptr, KernelCols * sizeof(float)); in Run() 122 packed_ptr += KernelCols * packed_matrix->layout.stride; in Run() 123 src_ptr += KernelCols; in Run() 129 (KernelCols - remaining_cols) * sizeof(float)); in Run() 135 #define RUY_USE_MEMCPY_ROWMAJOR_FLOAT_PACK(ThePath, KernelCols) \ argument 137 struct PackImpl<ThePath, FixedKernelLayout<Order::kRowMajor, 1, KernelCols>, \ [all …]
|
D | pack_arm.h | 577 template <typename Scalar, int KernelCols> 579 FixedKernelLayout<Order::kColMajor, 16, KernelCols>, Scalar, 588 RUY_DCHECK_EQ((end_col - start_col) % KernelCols, 0); 599 block_row * KernelCols; 605 kInputXor, KernelCols);
|