| /external/gemmlowp/public/ |
| D | gemmlowp.h | 35 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()
|
| D | map.h | 32 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/ |
| D | quantization_example.cc | 28 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;
|
| D | public.md | 36 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/gemmlowp/internal/ |
| D | 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,
|
| D | multi_thread_gemm.h | 442 const MatrixMap<const InputScalar, LhsOrder>& _lhs, in GemmWithPackedRhsTask()
444 MatrixMap<OutputScalar, ResultOrder>* _result, in GemmWithPackedRhsTask()
500 const MatrixMap<const InputScalar, LhsOrder> lhs;
502 MatrixMap<OutputScalar, ResultOrder> result;
609 const MatrixMap<const InputScalar, LhsOrder>& lhs, in MultiThreadGemm()
610 const MatrixMap<const InputScalar, RhsOrder>& rhs, in MultiThreadGemm()
611 MatrixMap<OutputScalar, ResultOrder>* result, in MultiThreadGemm()
|
| D | single_thread_gemm.h | 68 const MatrixMap<const InputScalar, LhsOrder>& lhs, in SingleThreadGemm()
69 const MatrixMap<const InputScalar, RhsOrder>& rhs, in SingleThreadGemm()
70 MatrixMap<OutputScalar, ResultOrder>* result, in SingleThreadGemm()
|
| D | unpack.h | 40 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()
180 MatrixMap<DstScalarType, MapOrder::ColMajor> dst_colmajor_map( in UnpackResult()
|
| D | 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,
|
| D | compute.h | 67 const MatrixMap<std::int32_t, MapOrder::ColMajor>& packed_result_block) { in MarkPackedResultBlockAsInitialized()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | linalg_ops_common.h | 115 using MatrixMap = Eigen::Map<Matrix>; variable
117 using MatrixMaps = gtl::InlinedVector<MatrixMap, 4>;
179 using MatrixMap = typename Base::MatrixMap; \
|
| D | cholesky_grad.cc | 34 using MatrixMap = typename Base::MatrixMap; typedef in tensorflow::CholeskyGrad
64 MatrixMap output_matrix = outputs->at(0); in ComputeMatrix()
|
| D | eigen_mkldnn_contraction_kernel_test.cc | 120 using MatrixMap = Map<Eigen::Matrix<Scalar, Dynamic, Dynamic, ColMajor>>; in TEST() typedef
122 MatrixMap lhs_mat(lhs.data(), m, k); in TEST()
123 MatrixMap rhs_mat(rhs.data(), k, n); in TEST()
|
| D | quantized_matmul_op.cc | 47 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()
|
| D | lu_op.cc | 41 using MatrixMap = Eigen::Map<Matrix>; typedef in tensorflow::LuOp
139 MatrixMap packed_triangular_factors( in ComputeTensorSlice()
|
| D | matrix_triangular_solve_op.cc | 84 MatrixMap& output = outputs->at(0); in ComputeMatrix()
176 MatrixMap& output = outputs->at(0); in ComputeMatrix()
|
| D | quantized_conv_ops.cc | 418 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 ()()
|
| D | deep_conv2d.cc | 192 MatrixMap; typedef
217 MatrixMap C(out_buffer, tile_spatial_size, in_stride); in operator ()()
688 MatrixMap; typedef
717 MatrixMap C(tile_transform, tile_spatial_size, coord_stride); in operator ()()
740 MatrixMap; typedef
766 MatrixMap C(out_transform_buffer, out_tile_spatial_size, out_buf_stride); in operator ()()
|
| /external/gemmlowp/test/ |
| D | test.cc | 131 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 …]
|
| D | test.h | 47 class Matrix : public MatrixMap<tScalar, tOrder> {
49 typedef MatrixMap<tScalar, tOrder> Map;
50 typedef MatrixMap<const tScalar, tOrder> ConstMap;
|
| D | benchmark_all_sizes.cc | 49 class Matrix : public MatrixMap<tScalar, tOrder> {
51 typedef MatrixMap<tScalar, tOrder> Map;
52 typedef MatrixMap<const tScalar, tOrder> ConstMap;
|
| /external/gemmlowp/eight_bit_int_gemm/ |
| D | eight_bit_int_gemm.cc | 91 MatrixMap<const std::uint8_t, LhsOrder> lhs(a, m, k, lda); in EightBitIntGemmImpl()
92 MatrixMap<const std::uint8_t, RhsOrder> rhs(b, k, n, ldb); in EightBitIntGemmImpl()
93 MatrixMap<std::uint8_t, ResultOrder> result(c, m, n, ldc); in EightBitIntGemmImpl()
126 MatrixMap<const std::uint8_t, LhsOrder> lhs(a, m, k, lda); in EightBitIntGemmInt32Impl()
127 MatrixMap<const std::uint8_t, RhsOrder> rhs(b, k, n, ldb); in EightBitIntGemmInt32Impl()
128 MatrixMap<std::int32_t, ResultOrder> result(c, m, n, ldc); in EightBitIntGemmInt32Impl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_matvec.h | 38 using MatrixMap = Eigen::Map<Matrix>; in MatVec() local
53 auto mat = MatrixMap(mat_buf, mat_rows, mat_cols); in MatVec()
|
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/kernels/ |
| D | hyperplane_lsh_probes.cc | 38 using MatrixMap = Eigen::Map<Matrix>; typedef in tensorflow::HyperplaneLSHProbesOp
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 124 using MatrixMap = typename std::conditional< variable
131 MatrixMap<Scalar> MapAsMatrixWithLastDimAsRows(Scalar* data, in MapAsMatrixWithLastDimAsRows()
136 return MatrixMap<Scalar>(data, rows, cols); in MapAsMatrixWithLastDimAsRows()
140 MatrixMap<Scalar> MapAsMatrixWithFirstDimAsCols(Scalar* data, in MapAsMatrixWithFirstDimAsCols()
144 return MatrixMap<Scalar>(data, rows, cols); in MapAsMatrixWithFirstDimAsCols()
148 MatrixMap<Scalar> MapAsMatrixWithFirstDimAsRows(Scalar* data, in MapAsMatrixWithFirstDimAsRows()
155 return MatrixMap<Scalar>(data, rows, cols); in MapAsMatrixWithFirstDimAsRows()
159 MatrixMap<Scalar> MapAsMatrixWithLastDimAsCols(Scalar* data, in MapAsMatrixWithLastDimAsCols()
166 return MatrixMap<Scalar>(data, rows, cols); in MapAsMatrixWithLastDimAsCols()
213 MatrixMap<Scalar> MapAsMatrixWithGivenNumberOfRows(Scalar* data, in MapAsMatrixWithGivenNumberOfRows()
[all …]
|