Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/lhlo_to_tfrt/
Dlhlo_gpu_to_tfrt_gpu.mlir48 lhs_stride = 20,
98 lhs_stride = 20,
147 lhs_stride = 20,
212 lhs_stride = 16,
226 lhs_stride = 16,
261 lhs_stride = 16,
287 lhs_stride = 16,
312 lhs_stride = 16,
/external/ruy/ruy/
Dkernel_common.h116 std::int32_t lhs_stride;
170 params->lhs_stride = lhs.layout.stride;
224 std::int32_t lhs_stride;
267 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;
686 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride;
769 const std::int64_t lhs_stride = params.lhs_stride >> 2;
777 params.lhs_base_ptr - params.start_row * lhs_stride;
790 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride;
842 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride;
Dkernel_avx512.cc610 lhs_col_ptr += 16 * params.lhs_stride;
850 lhs_col_ptr += 16 * params.lhs_stride;
858 const std::int64_t lhs_stride = params.lhs_stride >> 2;
871 params.lhs_base_ptr - params.start_row * lhs_stride;
886 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride;
1010 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride;
1151 const float* lhs_col_ptr = adj_lhs_col_ptr + row * lhs_stride;
1228 const std::int64_t lhs_stride = params.lhs_stride >> 2;
1235 params.lhs_base_ptr - params.start_row * lhs_stride;
1248 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()
7678 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/mlir/tfrt/tests/lhlo_to_jitrt/
Dgemm_to_jitrt.mlir43 lhs_stride = 16 : i64,
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lhlo_gpu_to_tfrt_gpu/
Dgemm_pattern.cc177 auto lhs_stride = builder.create<tfrt::compiler::ConstantI64Op>( in CreateTfrtOps() local
188 const_alpha, lhs_matrix.converted_value, data_type, lda, lhs_stride, in CreateTfrtOps()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/lhlo_gpu/
Dlhlo_gpu_ops.mlir202 lhs_stride = 20,
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContractionSycl.h640 EIGEN_CONSTEXPR StorageIndex lhs_stride =
653 lhs_index += lhs_stride;