Searched refs:lhs_stride (Results 1 – 9 of 9) sorted by relevance
/external/ruy/ruy/ |
D | kernel_common.h | 116 std::int32_t lhs_stride; 170 params->lhs_stride = lhs.layout.stride; 217 std::int32_t lhs_stride; 260 params->lhs_stride = sizeof(float) * lhs.layout.stride;
|
D | kernel_x86.h | 555 const std::int64_t lhs_stride = params.lhs_stride >> 2; 569 params.lhs_base_ptr - params.start_row * lhs_stride; 588 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride; 688 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride; 773 const std::int64_t lhs_stride = params.lhs_stride >> 2; 781 params.lhs_base_ptr - params.start_row * lhs_stride; 794 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride; 846 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride;
|
D | kernel_avx512.cc | 651 lhs_col_ptr += 16 * params.lhs_stride; 893 lhs_col_ptr += 16 * params.lhs_stride; 901 const std::int64_t lhs_stride = params.lhs_stride >> 2; 914 params.lhs_base_ptr - params.start_row * lhs_stride; 929 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride; 1226 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride; 1395 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride; 1472 const std::int64_t lhs_stride = params.lhs_stride >> 2; 1479 params.lhs_base_ptr - params.start_row * lhs_stride; 1492 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride; [all …]
|
D | kernel_arm32.cc | 68 static_assert(offsetof(Params, lhs_stride) == RUY_OFFSET_LHS_STRIDE, ""); in CheckOffsetsInKernelParamsFloat32() 618 static_assert(offsetof(Params, lhs_stride) == RUY_OFFSET_LHS_STRIDE, ""); in CheckOffsetsInKernelParams8bit()
|
D | kernel_avx2_fma.cc | 715 lhs_col_ptr += kAvx8bitBlockSize * params.lhs_stride; 987 lhs_col_ptr += kAvx8bitBlockSize * params.lhs_stride;
|
D | kernel_arm64.cc | 87 static_assert(offsetof(Params, lhs_stride) == RUY_OFFSET_LHS_STRIDE, ""); in CheckOffsetsInKernelParams8bit() 6423 static_assert(offsetof(Params, lhs_stride) == RUY_OFFSET_LHS_STRIDE, ""); in CheckOffsetsInKernelParamsFloat()
|
D | kernel_avx.cc | 1156 lhs_col_ptr += kAvx8bitBlockSize * params.lhs_stride; 1452 lhs_col_ptr += kAvx8bitBlockSize * params.lhs_stride;
|
/external/ruy/example/ |
D | parametrized_example.cc | 52 int lhs_stride; member 93 make_layout(params.m, params.k, params.lhs_stride, in run() 160 ¶ms.lhs_stride); in main()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gemm_thunk.cc | 145 int64 lhs_stride = lhs_matrix.num_rows * lhs_matrix.num_cols; in DoGemmWithAlgorithm() local 153 /*leading dim of LHS=*/lhs_matrix.num_rows, lhs_stride, rhs_data, in DoGemmWithAlgorithm()
|