• Home
  • Raw
  • Download

Lines Matching refs:MatrixMap

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()
243 const MatrixMap<const Scalar, RhsOrder>& rhs, in Gemm()
244 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset, in Gemm()
268 const MatrixMap<const Scalar, LhsOrder>& lhs, in Gemm()
269 const MatrixMap<const Scalar, RhsOrder>& rhs, in Gemm()
270 MatrixMap<Scalar, ResultOrder>* result, int lhs_offset, in 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()