Home
last modified time | relevance | path

Searched refs:lhs_stride (Results 1 – 9 of 9) sorted by relevance

/external/ruy/ruy/
Dkernel_common.h116 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;
Dkernel_x86.h555 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;
Dkernel_avx512.cc651 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 …]
Dkernel_arm32.cc68 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()
Dkernel_avx2_fma.cc715 lhs_col_ptr += kAvx8bitBlockSize * params.lhs_stride;
987 lhs_col_ptr += kAvx8bitBlockSize * params.lhs_stride;
Dkernel_arm64.cc87 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()
Dkernel_avx.cc1156 lhs_col_ptr += kAvx8bitBlockSize * params.lhs_stride;
1452 lhs_col_ptr += kAvx8bitBlockSize * params.lhs_stride;
/external/ruy/example/
Dparametrized_example.cc52 int lhs_stride; member
93 make_layout(params.m, params.k, params.lhs_stride, in run()
160 &params.lhs_stride); in main()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc145 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()