Searched refs:src_cols (Results 1 – 8 of 8) sorted by relevance
| /external/ruy/ruy/ |
| D | pack_common.h | 118 int src_cols = std::min(end_col, src_matrix.layout.cols) - start_col; in Run() local
120 for (; col <= src_cols - KernelCols; col += KernelCols) { in Run()
125 int remaining_cols = src_cols - col; in Run()
|
| D | pack_arm.cc | 1590 int src_inc3, int src_cols, in Pack8bitRowMajorForNeonDotprod() argument
1731 [src_cols] "r"(src_cols) in Pack8bitRowMajorForNeonDotprod()
2253 int src_rows, int src_cols, int block_row, in Pack8bitRowMajorForNeon() argument
2260 int src_end_col = std::min(end_col, src_cols); in Pack8bitRowMajorForNeon()
|
| D | pack_arm.h | 56 int src_rows, int src_cols, int block_row,
91 int src_inc3, int src_cols,
560 int src_cols = std::min(end_col, src_matrix.layout.cols) - start_col;
567 src_inc3, src_cols, src_matrix.zero_point, packed_ptr,
|
| D | pack_x86.h | 311 int src_cols, int block_row, int src_rows,
346 int src_cols, int block_row, int src_rows,
381 int src_cols, int block_row, int src_rows,
|
| D | pack_avx2_fma.cc | 609 int src_cols, int block_row, int src_rows,
612 int src_end_col = std::min(end_col, src_cols);
|
| D | pack_avx512.cc | 728 int src_cols, int block_row, int src_rows,
731 int src_end_col = std::min(end_col, src_cols);
|
| D | pack_avx.cc | 751 int src_cols, int block_row, int src_rows,
754 int src_end_col = std::min(end_col, src_cols);
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op_gpu.cu.cc | 189 const int src_cols) const { in TransposeWithGeam()
193 src_cols, &one, src, src_cols, &zero, in TransposeWithGeam()
|