Home
last modified time | relevance | path

Searched refs:VectorShape (Results 1 – 11 of 11) 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 const VectorShape TransposedShape = TransposeVectorShape<Shape>::Value;
Dsimd_wrappers.h310 VectorShape Shape>
316 static_assert(Shape == VectorShape::Col || Rows == 1, "");
317 static_assert(Shape == VectorShape::Row || Cols == 1, "");
327 VectorShape Shape>
333 static_assert(Shape == VectorShape::Col || Rows == 1, "");
334 static_assert(Shape == VectorShape::Row || Cols == 1, "");
377 VectorShape Shape>
380 static constexpr int kRows = Shape == VectorShape::Col ? BroadcastRows : 1;
381 static constexpr int kCols = Shape == VectorShape::Row ? BroadcastCols : 1;
385 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()
144 const VectorMap<const std::int32_t, VectorShape::Col> lhs_sums_of_each_slice( in UnpackResult()
146 const VectorMap<const std::int32_t, VectorShape::Row> rhs_sums_of_each_slice( in UnpackResult()
Doutput.h91 template <int Rows, int Cols, VectorShape Shape>
103 const int pos = Shape == VectorShape::Col ? row : col;
238 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 const VectorShape kShape = tShape;
112 template <typename tScalar, VectorShape tShape>
116 static const VectorShape kShape = tShape;
Doutput_stages.h61 template <VectorShape tShape>
194 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/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()
1287 typedef VectorMap<std::int32_t, VectorShape::Row> RowVectorMap; in TestOutputStages()
1308 typedef VectorMap<std::int32_t, VectorShape::Col> ColVectorMap; in TestOutputStages()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1139 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col>
1289 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col> in FullyConnected()
3782 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col> in LstmCell()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h1122 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col>
1297 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col> in FullyConnected()
3898 typedef gemmlowp::VectorMap<const int32, gemmlowp::VectorShape::Col> in LstmCell()