Home
last modified time | relevance | path

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

/external/gemmlowp/internal/
Ddispatch_gemm_shape.h47 template <VectorShape Shape>
49 static constexpr VectorShape Value =
50 Shape == VectorShape::Row ? VectorShape::Col : VectorShape::Row;
53 template <typename Scalar, VectorShape Shape>
56 static constexpr VectorShape TransposedShape =
74 template <VectorShape Shape>
77 static constexpr VectorShape TransposedShape =
89 template <VectorShape Shape>
92 static constexpr VectorShape TransposedShape =
Dsimd_wrappers.h461 VectorShape Shape>
467 static_assert(Shape == VectorShape::Col || Rows == 1, "");
468 static_assert(Shape == VectorShape::Row || Cols == 1, "");
478 VectorShape Shape>
484 static_assert(Shape == VectorShape::Col || Rows == 1, "");
485 static_assert(Shape == VectorShape::Row || Cols == 1, "");
528 VectorShape Shape>
531 static constexpr int kRows = Shape == VectorShape::Col ? BroadcastRows : 1;
532 static constexpr int kCols = Shape == VectorShape::Row ? BroadcastCols : 1;
536 VectorShape Shape>
[all …]
Dsimd_wrappers_common_neon_sse.h88 VectorMap<SrcScalarType, VectorShape::Col>> {
90 const VectorMap<SrcScalarType, VectorShape::Col>& src, int pos) {
99 VectorDup<SrcScalarType, VectorShape::Col>> {
101 const VectorDup<SrcScalarType, VectorShape::Col>& src, int) {
110 VectorMap<SrcScalarType, VectorShape::Col>> {
111 using SrcObjectType = VectorMap<SrcScalarType, VectorShape::Col>;
127 VectorMap<SrcScalarType, VectorShape::Col>> {
128 using SrcObjectType = VectorMap<SrcScalarType, VectorShape::Col>;
145 VectorMap<SrcScalarType, VectorShape::Row>> {
146 using SrcObjectType = VectorMap<SrcScalarType, VectorShape::Row>;
[all …]
Dunpack.h73 const VectorMap<const std::int32_t, VectorShape::Col>& in PrefetchResultBlock()
94 const VectorMap<const std::int32_t, VectorShape::Col>& in UnpackResultBlock()
96 const VectorMap<const std::int32_t, VectorShape::Row>& in UnpackResultBlock()
146 const VectorMap<const std::int32_t, VectorShape::Col> lhs_sums_of_each_slice( in UnpackResult()
148 const VectorMap<const std::int32_t, VectorShape::Row> rhs_sums_of_each_slice( in UnpackResult()
Doutput.h92 template <int Rows, int Cols, VectorShape Shape>
104 const int pos = Shape == VectorShape::Col ? row : col;
183 template <int Rows, int Cols, VectorShape Shape>
196 const int pos = Shape == VectorShape::Row ? col : row;
324 const int pos = VectorType::kShape == VectorShape::Row ? col : row;
/external/gemmlowp/public/
Dmap.h79 enum class VectorShape { Col, Row }; enum
83 template <typename tScalar, VectorShape tShape>
87 static constexpr VectorShape kShape = tShape;
112 template <typename tScalar, VectorShape tShape>
116 static constexpr VectorShape kShape = tShape;
Doutput_stages.h61 template <VectorShape tShape>
154 template <VectorShape tShape>
226 template <VectorShape tShape>
Dgemmlowp.h59 typedef VectorDup<const std::int32_t, VectorShape::Col> OffsetColDup; in GemmWithOutputPipeline()
60 typedef VectorDup<const std::int32_t, VectorShape::Row> OffsetRowDup; in GemmWithOutputPipeline()
/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm_gemmlowp.h96 gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col>;
156 gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col>;
161 gemmlowp::VectorShape::Col>
/external/gemmlowp/test/
Dtest.cc109 typedef VectorMap<const std::int32_t, VectorShape::Col> OffsetColMap;
110 typedef VectorMap<const std::int32_t, VectorShape::Row> OffsetRowMap;
111 typedef VectorDup<const std::int32_t, VectorShape::Col> OffsetColDup;
112 typedef VectorDup<const std::int32_t, VectorShape::Row> OffsetRowDup;
780 auto output_pipeline = MakeStandardOutputPipeline<VectorShape::Col>( in TestWithSmallDataPerChannelQuantization()
977 auto output_pipeline = MakeStandardOutputPipeline<VectorShape::Col>( in TestWithLargeDataPerChannelQuantization()
1090 auto output_pipeline = MakeStandardOutputPipeline<VectorShape::Col>( in TestMultithreadedPerChannelQuantization()
1328 typedef VectorMap<std::int32_t, VectorShape::Row> RowVectorMap; in TestOutputStages()
1349 typedef VectorMap<std::int32_t, VectorShape::Col> ColVectorMap; in TestOutputStages()
/external/XNNPACK/bench/
Dqu8-gemm.cc112 typedef gemmlowp::VectorMap<const int32_t, gemmlowp::VectorShape::Col> ColVectorMap;
/external/ruy/ruy/
Dtest.h808 gemmlowp::VectorShape::Col>
812 gemmlowp::VectorMap<const std::int32_t, gemmlowp::VectorShape::Col>;
831 gemmlowp::VectorMap<const std::int32_t, gemmlowp::VectorShape::Col>;
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h715 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col>
743 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col>
1698 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col> in FullyConnected()
3228 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col> in LstmCell()