Home
last modified time | relevance | path

Searched refs:MatrixMap (Results 1 – 25 of 39) sorted by relevance

12

/external/gemmlowp/public/
Dgemmlowp.h35 const MatrixMap<const InputScalar, LhsOrder>& lhs, in GemmWithOutputPipelinePC()
36 const MatrixMap<const InputScalar, RhsOrder>& rhs, in GemmWithOutputPipelinePC()
37 MatrixMap<OutputScalar, ResultOrder>* result, in GemmWithOutputPipelinePC()
54 const MatrixMap<const InputScalar, LhsOrder>& lhs, in GemmWithOutputPipeline()
55 const MatrixMap<const InputScalar, RhsOrder>& rhs, in GemmWithOutputPipeline()
56 MatrixMap<OutputScalar, ResultOrder>* result, in GemmWithOutputPipeline()
75 const MatrixMap<const Scalar, LhsOrder>& lhs, in Gemm()
76 const MatrixMap<const Scalar, RhsOrder>& rhs, in Gemm()
77 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset, in Gemm()
Dmap.h32 class MatrixMap {
42 MatrixMap() : data_(nullptr), rows_(0), cols_(0), stride_(0) {} in MatrixMap() function
43 MatrixMap(Scalar* data, int rows, int cols) in MatrixMap() function
48 MatrixMap(Scalar* data, int rows, int cols, int stride) in MatrixMap() function
50 MatrixMap(const MatrixMap& other) in MatrixMap() function
67 MatrixMap block(int start_row, int start_col, int block_rows, in block()
74 return MatrixMap(data(start_row, start_col), block_rows, block_cols, in block()
/external/gemmlowp/doc/
Dquantization_example.cc28 const gemmlowp::MatrixMap<tScalar, tOrder>& m) { in operator <<()
43 void FindMinMax(const gemmlowp::MatrixMap<float, tOrder>& m, float* min, in FindMinMax()
117 const gemmlowp::MatrixMap<const float, tLhsOrder>& lhs, in FloatMatrixMultiplication()
118 const gemmlowp::MatrixMap<const float, tRhsOrder>& rhs, in FloatMatrixMultiplication()
119 gemmlowp::MatrixMap<float, tResultOrder>* result) { in FloatMatrixMultiplication()
165 gemmlowp::MatrixMap<const tScalar, tOrder> ConstMap() const { in ConstMap()
166 return gemmlowp::MatrixMap<const tScalar, tOrder>( in ConstMap()
169 gemmlowp::MatrixMap<tScalar, tOrder> Map() { in Map()
170 return gemmlowp::MatrixMap<tScalar, tOrder>( in Map()
178 gemmlowp::MatrixMap<tScalar, tOrder> matrix_map;
Dpublic.md36 const MatrixMap<const InputScalar, LhsOrder>& lhs,
37 const MatrixMap<const InputScalar, RhsOrder>& rhs,
38 MatrixMap<OutputScalar, ResultOrder>* result,
87 `MatrixMap` objects, mapping external buffers as matrices, not owning data.
89 * `result`: pointer to the destination `MatrixMap` object, which must be
/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm_gemmlowp.h88 gemmlowp::MatrixMap<const SrcScalar, gemmlowp::MapOrder::RowMajor>
90 gemmlowp::MatrixMap<const SrcScalar, gemmlowp::MapOrder::ColMajor>
92 gemmlowp::MatrixMap<DstScalar, gemmlowp::MapOrder::ColMajor> gemmlowp_dst(
148 gemmlowp::MatrixMap<const SrcScalar, gemmlowp::MapOrder::RowMajor>
150 gemmlowp::MatrixMap<const SrcScalar, gemmlowp::MapOrder::ColMajor>
152 gemmlowp::MatrixMap<DstScalar, gemmlowp::MapOrder::ColMajor> gemmlowp_dst(
/external/gemmlowp/internal/
Ddispatch_gemm_shape.h65 struct TransposeImpl<MatrixMap<Scalar, Order>> {
66 typedef MatrixMap<Scalar, Order> SrcType;
68 typedef MatrixMap<Scalar, TransposedOrder> DstType;
174 const MatrixMap<const InputScalar, LhsOrder>& lhs,
175 const MatrixMap<const InputScalar, RhsOrder>& rhs,
176 MatrixMap<OutputScalar, ResultOrder>* result,
Dsingle_thread_gemm.h68 const MatrixMap<const InputScalar, LhsOrder>& lhs, in SingleThreadGemm()
69 const MatrixMap<const InputScalar, RhsOrder>& rhs, in SingleThreadGemm()
70 MatrixMap<OutputScalar, ResultOrder>* result, in SingleThreadGemm()
Dmulti_thread_gemm.h470 const MatrixMap<const InputScalar, LhsOrder>& _lhs, in GemmWithPackedRhsTask()
472 MatrixMap<OutputScalar, ResultOrder>* _result, in GemmWithPackedRhsTask()
528 const MatrixMap<const InputScalar, LhsOrder> lhs;
530 MatrixMap<OutputScalar, ResultOrder> result;
637 const MatrixMap<const InputScalar, LhsOrder>& lhs, in MultiThreadGemm()
638 const MatrixMap<const InputScalar, RhsOrder>& rhs, in MultiThreadGemm()
639 MatrixMap<OutputScalar, ResultOrder>* result, in MultiThreadGemm()
Dunpack.h40 MatrixMap<std::int32_t, MapOrder::ColMajor> Map() { in Map()
41 return MatrixMap<std::int32_t, MapOrder::ColMajor>( in Map()
46 MatrixMap<const std::int32_t, MapOrder::ColMajor> Map() const { in Map()
47 return MatrixMap<const std::int32_t, MapOrder::ColMajor>( in Map()
182 MatrixMap<DstScalarType, MapOrder::ColMajor> dst_colmajor_map( in UnpackResult()
Dsimd_wrappers_common_neon_sse.h26 MatrixMap<SrcScalarType, MapOrder::ColMajor>> {
28 const MatrixMap<SrcScalarType, MapOrder::ColMajor>& src, int row,
40 MatrixMap<SrcScalarType, MapOrder::ColMajor>> {
42 const MatrixMap<SrcScalarType, MapOrder::ColMajor>& src, int row,
55 MatrixMap<SrcScalarType, MapOrder::ColMajor>> {
57 const MatrixMap<SrcScalarType, MapOrder::ColMajor>& src, int row,
71 MatrixMap<SrcScalarType, MapOrder::ColMajor>> {
73 const MatrixMap<SrcScalarType, MapOrder::ColMajor>& src, int row,
/external/tensorflow/tensorflow/core/kernels/linalg/
Dtridiagonal_matmul_op.cc99 MatrixMap& result = outputs->at(0); in ComputeMatrix()
109 MatrixMap result_except_first_row(result.data() + n, m - 1, n); in ComputeMatrix()
110 MatrixMap result_except_last_row(result.data(), m - 1, n); in ComputeMatrix()
Dlinalg_ops_common.h136 using MatrixMap = Eigen::Map<Matrix>; variable
140 using MatrixMaps = gtl::InlinedVector<MatrixMap, 4>;
202 using MatrixMap = typename Base::MatrixMap; \
Dcholesky_grad.cc34 using MatrixMap = typename Base::MatrixMap; typedef in tensorflow::CholeskyGrad
64 MatrixMap output_matrix = outputs->at(0); in ComputeMatrix()
Dtridiagonal_solve_op.cc114 MatrixMap& x = outputs->at(0); in ComputeMatrix()
143 MatrixMap& x) { in SolveWithGaussianEliminationWithPivoting()
198 const ConstMatrixMap& rhs, MatrixMap& x) { in SolveWithThomasAlgorithm()
Dlu_op.cc41 using MatrixMap = Eigen::Map<Matrix>; typedef in tensorflow::LuOp
139 MatrixMap packed_triangular_factors( in ComputeTensorSlice()
Dbanded_triangular_solve_op.cc46 using MatrixMap = Eigen::Map<Matrix>; typedef
56 static MatrixMap TensorSliceToEigenMatrix(Tensor* t, int slice) { in TensorSliceToEigenMatrix()
57 return MatrixMap( in TensorSliceToEigenMatrix()
Dmatrix_triangular_solve_op_impl.h68 using MatrixMap = Eigen::Map<Matrix>; member
78 static MatrixMap TensorSliceToEigenMatrix(Tensor* t, int slice) { in TensorSliceToEigenMatrix()
79 return MatrixMap( in TensorSliceToEigenMatrix()
/external/gemmlowp/test/
Dtest.cc131 const MatrixMap<const Scalar, LhsOrder>& lhs, in Gemm()
132 const MatrixMap<const Scalar, RhsOrder>& rhs, in Gemm()
133 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset, in Gemm()
171 const MatrixMap<const Scalar, LhsOrder>& lhs, in Gemm()
172 const MatrixMap<const Scalar, RhsOrder>& rhs, in Gemm()
173 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset, in Gemm()
208 const MatrixMap<const Scalar, LhsOrder>& lhs, in Gemm()
209 const MatrixMap<const Scalar, RhsOrder>& rhs, in Gemm()
210 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset, in Gemm()
242 static bool Gemm(Context*, const MatrixMap<const Scalar, LhsOrder>& lhs, in Gemm()
[all …]
Dtest.h47 class Matrix : public MatrixMap<tScalar, tOrder> {
49 typedef MatrixMap<tScalar, tOrder> Map;
50 typedef MatrixMap<const tScalar, tOrder> ConstMap;
Dbenchmark_all_sizes.cc53 class Matrix : public MatrixMap<tScalar, tOrder> {
55 typedef MatrixMap<tScalar, tOrder> Map;
56 typedef MatrixMap<const tScalar, tOrder> ConstMap;
/external/gemmlowp/eight_bit_int_gemm/
Deight_bit_int_gemm.cc88 MatrixMap<const std::uint8_t, LhsOrder> lhs(a, m, k, lda); in EightBitIntGemmImpl()
89 MatrixMap<const std::uint8_t, RhsOrder> rhs(b, k, n, ldb); in EightBitIntGemmImpl()
90 MatrixMap<std::uint8_t, ResultOrder> result(c, m, n, ldc); in EightBitIntGemmImpl()
123 MatrixMap<const std::uint8_t, LhsOrder> lhs(a, m, k, lda); in EightBitIntGemmInt32Impl()
124 MatrixMap<const std::uint8_t, RhsOrder> rhs(b, k, n, ldb); in EightBitIntGemmInt32Impl()
125 MatrixMap<std::int32_t, ResultOrder> result(c, m, n, ldc); in EightBitIntGemmInt32Impl()
/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc128 using MatrixMap = Map<Eigen::Matrix<Scalar, Dynamic, Dynamic, ColMajor>>; in TEST() typedef
130 MatrixMap lhs_mat(lhs.data(), m, k); in TEST()
131 MatrixMap rhs_mat(rhs.data(), k, n); in TEST()
Dquantized_matmul_op.cc47 gemmlowp::MatrixMap<const std::uint8_t, LhsOrder> lhs(a_data_as_uint8, m, k, in GemmlowpMultiply()
49 gemmlowp::MatrixMap<const std::uint8_t, RhsOrder> rhs(b_data_as_uint8, k, n, in GemmlowpMultiply()
51 gemmlowp::MatrixMap<std::int32_t, ResultOrder> result(c_data_as_int32, m, n, in GemmlowpMultiply()
Dquantized_conv_ops.cc418 gemmlowp::MatrixMap<const std::uint8_t, LhsOrder> lhs( in operator ()()
420 gemmlowp::MatrixMap<const std::uint8_t, RhsOrder> rhs( in operator ()()
422 gemmlowp::MatrixMap<std::int32_t, ResultOrder> result( in operator ()()
/external/XNNPACK/bench/
Dqu8-gemm.cc182 gemmlowp::MatrixMap<const uint8_t, gemmlowp::MapOrder::RowMajor> AM(a.data(), mc, kc, kc); in GemmlowpBenchmark()
183 …gemmlowp::MatrixMap<const uint8_t, gemmlowp::MapOrder::ColMajor> BM(k.data() + buffer_index * kEle… in GemmlowpBenchmark()
184 …gemmlowp::MatrixMap<uint8_t, gemmlowp::MapOrder::RowMajor> CM(c.data() + buffer_index * c_elements… in GemmlowpBenchmark()

12