Lines Matching refs:MapOrder
120 template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
151 template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
178 template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
210 template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
216 const bool transpose_c = ResultOrder == MapOrder::RowMajor; in Gemm()
217 const bool transpose_a = LhsOrder == MapOrder::RowMajor; in Gemm()
218 const bool transpose_b = RhsOrder == MapOrder::RowMajor; in Gemm()
233 template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
248 const char* OrderName(MapOrder order) { in OrderName()
249 return order == MapOrder::ColMajor ? "ColMajor" : "RowMajor"; in OrderName()
381 static const MapOrder kLhsOrder = LhsType::kOrder; in test_gemm_impl()
382 static const MapOrder kRhsOrder = RhsType::kOrder; in test_gemm_impl()
383 static const MapOrder kResultOrder = ResultType::kOrder; in test_gemm_impl()
385 const bool transpose_c = kResultOrder == MapOrder::RowMajor; in test_gemm_impl()
386 const bool transpose_a = kLhsOrder == MapOrder::RowMajor; in test_gemm_impl()
387 const bool transpose_b = kRhsOrder == MapOrder::RowMajor; in test_gemm_impl()
484 template <typename GemmWrapper, MapOrder LhsOrder, MapOrder RhsOrder,
485 MapOrder ResultOrder>
519 test_gemm<GemmWrapper, MapOrder::LhsOrder, MapOrder::RhsOrder, \ in test_gemm()
520 MapOrder::ResultOrder>(context, rows, depth, cols, \ in test_gemm()
722 MatrixMap<const std::uint8_t, MapOrder::RowMajor> lhs(a_data, m, k, lda); in TestWithSmallDataPerChannelQuantization()
739 MatrixMap<const std::uint8_t, MapOrder::ColMajor> rhs(b_data, k, n, ldb); in TestWithSmallDataPerChannelQuantization()
761 MatrixMap<std::uint8_t, MapOrder::ColMajor> result(output_data.get(), m, n, in TestWithSmallDataPerChannelQuantization()
848 MatrixMap<const std::uint8_t, MapOrder::RowMajor> lhs(a_data, m, k, lda); in TestWithLargeDataPerChannelQuantization()
906 MatrixMap<const std::uint8_t, MapOrder::ColMajor> rhs(b_data, k, n, ldb); in TestWithLargeDataPerChannelQuantization()
981 MatrixMap<std::uint8_t, MapOrder::ColMajor> result(output_data.get(), m, n, in TestWithLargeDataPerChannelQuantization()
1096 template <MapOrder ResultOrder>
1099 Matrix<std::uint8_t, MapOrder::RowMajor> lhs(rows, depth); in TestOutputStages()
1100 Matrix<std::uint8_t, MapOrder::ColMajor> rhs(depth, cols); in TestOutputStages()
1431 TestOutputStages<MapOrder::RowMajor>(63, 10, 127, 5, 17, 14); in test()
1432 TestOutputStages<MapOrder::ColMajor>(63, 10, 127, 5, 17, 14); in test()