| /external/gemmlowp/public/ |
| gemmlowp.h | 35 const MatrixMap<const InputScalar, LhsOrder>& lhs,
36 const MatrixMap<const InputScalar, RhsOrder>& rhs,
37 MatrixMap<OutputScalar, ResultOrder>* result,
54 const MatrixMap<const InputScalar, LhsOrder>& lhs,
55 const MatrixMap<const InputScalar, RhsOrder>& rhs,
56 MatrixMap<OutputScalar, ResultOrder>* result,
75 const MatrixMap<const Scalar, LhsOrder>& lhs,
76 const MatrixMap<const Scalar, RhsOrder>& rhs,
77 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset,
|
| map.h | 29 // A MatrixMap is a view of an existing buffer as a matrix. It does not own
32 class MatrixMap {
42 MatrixMap() : data_(nullptr), rows_(0), cols_(0), stride_(0) {}
43 MatrixMap(Scalar* data, int rows, int cols)
48 MatrixMap(Scalar* data, int rows, int cols, int stride)
50 MatrixMap(const MatrixMap& other)
67 MatrixMap block(int start_row, int start_col, int block_rows,
74 return MatrixMap(data(start_row, start_col), block_rows, block_cols,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| linalg_ops_common.h | 115 using MatrixMap = Eigen::Map<Matrix>;
117 using MatrixMaps = gtl::InlinedVector<MatrixMap, 4>;
179 using MatrixMap = typename Base::MatrixMap; \
|
| cholesky_grad.cc | 34 using MatrixMap = typename Base::MatrixMap;
64 MatrixMap output_matrix = outputs->at(0);
|
| quantized_matmul_op.cc | 47 gemmlowp::MatrixMap<const std::uint8_t, LhsOrder> lhs(a_data_as_uint8, m, k,
49 gemmlowp::MatrixMap<const std::uint8_t, RhsOrder> rhs(b_data_as_uint8, k, n,
51 gemmlowp::MatrixMap<std::int32_t, ResultOrder> result(c_data_as_int32, m, n,
|
| matrix_triangular_solve_op.cc | 86 MatrixMap& output = outputs->at(0);
178 MatrixMap& output = outputs->at(0);
|
| batch_matmul_op_impl.h | 155 using MatrixMap = Eigen::Map<Matrix>;
164 static MatrixMap TensorSliceToEigenMatrix(Tensor* t, int slice) {
165 return MatrixMap(
|
| quantized_conv_ops.cc | 418 gemmlowp::MatrixMap<const std::uint8_t, LhsOrder> lhs(
420 gemmlowp::MatrixMap<const std::uint8_t, RhsOrder> rhs(
422 gemmlowp::MatrixMap<std::int32_t, ResultOrder> result(
|
| deep_conv2d.cc | 192 MatrixMap;
217 MatrixMap C(out_buffer, tile_spatial_size, in_stride);
687 MatrixMap;
716 MatrixMap C(tile_transform, tile_spatial_size, coord_stride);
739 MatrixMap;
765 MatrixMap C(out_transform_buffer, out_tile_spatial_size, out_buf_stride);
[all...] |
| sparse_matmul_op.cc | 60 using MatrixMap = BasicMatrixMap<float>;
790 bool transpose_output, MatrixMap* output);
800 bool transpose_output, MatrixMap* output);
[all...] |
| /external/gemmlowp/doc/ |
| quantization_example.cc | 22 // represent as gemmlowp::MatrixMap.
28 const gemmlowp::MatrixMap<tScalar, tOrder>& m) {
43 void FindMinMax(const gemmlowp::MatrixMap<float, tOrder>& m, float* min,
117 const gemmlowp::MatrixMap<const float, tLhsOrder>& lhs,
118 const gemmlowp::MatrixMap<const float, tRhsOrder>& rhs,
119 gemmlowp::MatrixMap<float, tResultOrder>* result) {
165 gemmlowp::MatrixMap<const tScalar, tOrder> ConstMap() const {
166 return gemmlowp::MatrixMap<const tScalar, tOrder>(
169 gemmlowp::MatrixMap<tScalar, tOrder> Map() {
170 return gemmlowp::MatrixMap<tScalar, tOrder>
[all...] |
| /external/gemmlowp/internal/ |
| single_thread_gemm.h | 68 const MatrixMap<const InputScalar, LhsOrder>& lhs,
69 const MatrixMap<const InputScalar, RhsOrder>& rhs,
70 MatrixMap<OutputScalar, ResultOrder>* result,
|
| dispatch_gemm_shape.h | 65 struct TransposeImpl<MatrixMap<Scalar, Order>> {
66 typedef MatrixMap<Scalar, Order> SrcType;
68 typedef MatrixMap<Scalar, TransposedOrder> DstType;
156 const MatrixMap<const InputScalar, LhsOrder>& lhs,
157 const MatrixMap<const InputScalar, RhsOrder>& rhs,
158 MatrixMap<OutputScalar, ResultOrder>* result,
|
| unpack.h | 40 MatrixMap<std::int32_t, MapOrder::ColMajor> Map() {
41 return MatrixMap<std::int32_t, MapOrder::ColMajor>(
46 MatrixMap<const std::int32_t, MapOrder::ColMajor> Map() const {
47 return MatrixMap<const std::int32_t, MapOrder::ColMajor>(
180 MatrixMap<DstScalarType, MapOrder::ColMajor> dst_colmajor_map(
|
| multi_thread_gemm.h | 442 const MatrixMap<const InputScalar, LhsOrder>& _lhs,
444 MatrixMap<OutputScalar, ResultOrder>* _result,
500 const MatrixMap<const InputScalar, LhsOrder> lhs;
502 MatrixMap<OutputScalar, ResultOrder> result;
609 const MatrixMap<const InputScalar, LhsOrder>& lhs,
610 const MatrixMap<const InputScalar, RhsOrder>& rhs,
611 MatrixMap<OutputScalar, ResultOrder>* result,
|
| simd_wrappers_common_neon_sse.h | 26 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,
|
| compute.h | 67 const MatrixMap<std::int32_t, MapOrder::ColMajor>& packed_result_block) {
|
| simd_wrappers.h | 293 MatrixMap<SrcScalarType, MapOrder::ColMajor>> {
295 using SrcObjectType = MatrixMap<SrcScalarType, MapOrder::ColMajor>;
|
| /external/gemmlowp/test/ |
| test.h | 43 // gemmlowp itself doesn't have a Matrix class, only a MatrixMap class,
47 class Matrix : public MatrixMap<tScalar, tOrder> {
49 typedef MatrixMap<tScalar, tOrder> Map;
50 typedef MatrixMap<const tScalar, tOrder> ConstMap;
|
| test.cc | 131 const MatrixMap<const Scalar, LhsOrder>& lhs,
132 const MatrixMap<const Scalar, RhsOrder>& rhs,
133 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset,
171 const MatrixMap<const Scalar, LhsOrder>& lhs,
172 const MatrixMap<const Scalar, RhsOrder>& rhs,
173 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset,
208 const MatrixMap<const Scalar, LhsOrder>& lhs,
209 const MatrixMap<const Scalar, RhsOrder>& rhs,
210 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset,
242 static bool Gemm(Context*, const MatrixMap<const Scalar, LhsOrder>& lhs
[all...] |
| benchmark_all_sizes.cc | 45 // gemmlowp itself doesn't have a Matrix class, only a MatrixMap class,
49 class Matrix : public MatrixMap<tScalar, tOrder> {
51 typedef MatrixMap<tScalar, tOrder> Map;
52 typedef MatrixMap<const tScalar, tOrder> ConstMap;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_matvec.cc | 35 using MatrixMap = Eigen::Map<Matrix>;
50 auto mat = MatrixMap(mat_buf, mat_rows, mat_cols);
|
| /external/gemmlowp/eight_bit_int_gemm/ |
| eight_bit_int_gemm.cc | 91 MatrixMap<const std::uint8_t, LhsOrder> lhs(a, m, k, lda);
92 MatrixMap<const std::uint8_t, RhsOrder> rhs(b, k, n, ldb);
93 MatrixMap<std::uint8_t, ResultOrder> result(c, m, n, ldc);
126 MatrixMap<const std::uint8_t, LhsOrder> lhs(a, m, k, lda);
127 MatrixMap<const std::uint8_t, RhsOrder> rhs(b, k, n, ldb);
128 MatrixMap<std::int32_t, ResultOrder> result(c, m, n, ldc);
|
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/kernels/ |
| hyperplane_lsh_probes.cc | 38 using MatrixMap = Eigen::Map<Matrix>;
|
| hyperplane_lsh_probes.h | 41 using MatrixMap = Eigen::Map<Matrix>;
|
