/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);
|
eigen_mkldnn_contraction_kernel_test.cc | 120 using MatrixMap = Map<Eigen::Matrix<Scalar, Dynamic, Dynamic, ColMajor>>; 122 MatrixMap lhs_mat(lhs.data(), m, k); 123 MatrixMap rhs_mat(rhs.data(), k, n);
|
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,
|
lu_op.cc | 41 using MatrixMap = Eigen::Map<Matrix>; 139 MatrixMap packed_triangular_factors(
|
matrix_triangular_solve_op.cc | 84 MatrixMap& output = outputs->at(0); 176 MatrixMap& output = outputs->at(0);
|
tridiagonal_solve_op.cc | 96 MatrixMap& x = outputs->at(0);
|
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); 688 MatrixMap; 717 MatrixMap C(tile_transform, tile_spatial_size, coord_stride); 740 MatrixMap; 766 MatrixMap C(out_transform_buffer, out_tile_spatial_size, out_buf_stride); [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) {
|
/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.h | 38 using MatrixMap = Eigen::Map<Matrix>; 53 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>;
|