• Home
  • Raw
  • Download

Lines Matching refs:MapOrder

129   template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
169 template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
206 template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
241 template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
248 const bool transpose_c = ResultOrder == MapOrder::RowMajor; in Gemm()
249 const bool transpose_a = LhsOrder == MapOrder::RowMajor; in Gemm()
250 const bool transpose_b = RhsOrder == MapOrder::RowMajor; in Gemm()
266 template <MapOrder LhsOrder, MapOrder RhsOrder, MapOrder ResultOrder>
283 const char* OrderName(MapOrder order) { in OrderName()
284 return order == MapOrder::ColMajor ? "ColMajor" : "RowMajor"; in OrderName()
422 static const MapOrder kLhsOrder = LhsType::kOrder; in test_gemm_impl()
423 static const MapOrder kRhsOrder = RhsType::kOrder; in test_gemm_impl()
424 static const MapOrder kResultOrder = ResultType::kOrder; in test_gemm_impl()
426 const bool transpose_c = kResultOrder == MapOrder::RowMajor; in test_gemm_impl()
427 const bool transpose_a = kLhsOrder == MapOrder::RowMajor; in test_gemm_impl()
428 const bool transpose_b = kRhsOrder == MapOrder::RowMajor; in test_gemm_impl()
506 template <typename GemmWrapper, MapOrder LhsOrder, MapOrder RhsOrder,
507 MapOrder ResultOrder>
542 test_gemm<GemmWrapper, MapOrder::LhsOrder, MapOrder::RhsOrder, \ in test_gemm()
543 MapOrder::ResultOrder>(context, rows, depth, cols, \ in test_gemm()
744 MatrixMap<const std::uint8_t, MapOrder::RowMajor> lhs(a_data, m, k, lda); in TestWithSmallDataPerChannelQuantization()
759 MatrixMap<const std::uint8_t, MapOrder::ColMajor> rhs(b_data, k, n, ldb); in TestWithSmallDataPerChannelQuantization()
773 MatrixMap<std::uint8_t, MapOrder::ColMajor> result(output_data.get(), m, n, in TestWithSmallDataPerChannelQuantization()
854 MatrixMap<const std::uint8_t, MapOrder::RowMajor> lhs(a_data, m, k, lda); in TestWithLargeDataPerChannelQuantization()
910 MatrixMap<const std::uint8_t, MapOrder::ColMajor> rhs(b_data, k, n, ldb); in TestWithLargeDataPerChannelQuantization()
970 MatrixMap<std::uint8_t, MapOrder::ColMajor> result(output_data.get(), m, n, in TestWithLargeDataPerChannelQuantization()
1041 MatrixMap<const std::uint8_t, MapOrder::RowMajor> lhs(&a_data[0], m, k, lda); in TestMultithreadedPerChannelQuantization()
1052 MatrixMap<const std::uint8_t, MapOrder::ColMajor> rhs(&b_data[0], k, n, ldb); in TestMultithreadedPerChannelQuantization()
1083 MatrixMap<std::uint8_t, MapOrder::ColMajor> result(output_data.get(), m, n, in TestMultithreadedPerChannelQuantization()
1200 template <typename BitDepthParams, MapOrder ResultOrder>
1203 Matrix<std::uint8_t, MapOrder::RowMajor> lhs(rows, depth); in TestOutputStages()
1204 Matrix<std::uint8_t, MapOrder::ColMajor> rhs(depth, cols); in TestOutputStages()
1580 TestOutputStages<BitDepthParams, MapOrder::RowMajor>(63, 10, 127, 5, 17, 14); in TestOutputStages()
1581 TestOutputStages<BitDepthParams, MapOrder::ColMajor>(63, 10, 127, 5, 17, 14); in TestOutputStages()
1582 TestOutputStages<BitDepthParams, MapOrder::RowMajor>(630, 10, 1270, 5, 17, in TestOutputStages()
1584 TestOutputStages<BitDepthParams, MapOrder::ColMajor>(630, 10, 1270, 5, 17, in TestOutputStages()