Searched refs:rhs_stride (Results 1 – 9 of 9) sorted by relevance
| /external/ruy/ruy/ |
| D | kernel_common.h | 117 std::int32_t rhs_stride;
171 params->rhs_stride = rhs.layout.stride;
218 std::int32_t rhs_stride;
261 params->rhs_stride = sizeof(float) * rhs.layout.stride;
|
| D | kernel_x86.h | 557 const std::int64_t rhs_stride = params.rhs_stride >> 2;
565 params.rhs_base_ptr - params.start_col * rhs_stride;
582 const float* rhs_col_ptr = adj_rhs_col_ptr + col * rhs_stride;
681 const float* rhs_col_ptr = adj_rhs_col_ptr + col * rhs_stride;
|
| D | kernel_avx512.cc | 656 rhs_col_ptr += 16 * params.rhs_stride;
903 const std::int64_t rhs_stride = params.rhs_stride >> 2;
910 params.rhs_base_ptr - params.start_col * rhs_stride;
924 const float* rhs_col_ptr = adj_rhs_col_ptr + col * rhs_stride;
1389 const float* rhs_col_ptr = adj_rhs_col_ptr + col * rhs_stride;
|
| D | kernel_arm32.cc | 69 static_assert(offsetof(Params, rhs_stride) == RUY_OFFSET_RHS_STRIDE, ""); in CheckOffsetsInKernelParamsFloat32()
619 static_assert(offsetof(Params, rhs_stride) == RUY_OFFSET_RHS_STRIDE, ""); in CheckOffsetsInKernelParams8bit()
|
| D | kernel_avx2_fma.cc | 720 rhs_col_ptr += kAvx8bitBlockSize * params.rhs_stride;
992 rhs_col_ptr += kAvx8bitBlockSize * params.rhs_stride;
|
| D | kernel_arm64.cc | 88 static_assert(offsetof(Params, rhs_stride) == RUY_OFFSET_RHS_STRIDE, ""); in CheckOffsetsInKernelParams8bit()
6424 static_assert(offsetof(Params, rhs_stride) == RUY_OFFSET_RHS_STRIDE, ""); in CheckOffsetsInKernelParamsFloat()
|
| D | kernel_avx.cc | 1161 rhs_col_ptr += kAvx8bitBlockSize * params.rhs_stride;
1457 rhs_col_ptr += kAvx8bitBlockSize * params.rhs_stride;
|
| /external/ruy/example/ |
| D | parametrized_example.cc | 53 int rhs_stride; member
95 make_layout(params.k, params.n, params.rhs_stride, in run()
162 ¶ms.rhs_stride); in main()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 146 int64 rhs_stride = rhs_matrix.num_rows * rhs_matrix.num_cols; in DoGemmWithAlgorithm() local
154 /*leading dim of RHS=*/rhs_matrix.num_rows, rhs_stride, in DoGemmWithAlgorithm()
|