Lines Matching refs:cols
139 const int cols = rhs.cols();
140 if (rows < cols) {
141 // SingleThreadGemm is never called with rows < cols.
146 const OffsetRowDup rhs_offset_vector(rhs_offset, cols);
180 const int cols = rhs.cols();
181 if (rows < cols) {
182 // SingleThreadGemm is never called with rows < cols.
187 const OffsetRowDup rhs_offset_vector(rhs_offset, cols);
253 transpose_a, transpose_b, transpose_c, lhs.rows(), rhs.cols(),
254 lhs.cols(), lhs.data(), lhs_offset, lhs.stride(), rhs.data(),
276 rhs.cols(), lhs.cols(), lhs.data(), lhs_offset,
407 const int cols = rhs.cols();
408 Check(lhs.cols() == rhs.rows());
409 const int depth = lhs.cols();
417 // (e.g. rows < cols) as these are supposed to have been handled
426 ResultType ref_result(rows, cols);
438 GetResultStats(result->data(), ref_result.data(), rows * cols, &stats);
471 good ? "PASS" : "FAIL", rows, depth, cols, OrderName(kLhsOrder),
479 for (int c = 0; c < result->cols() && bad_coeffs_printed < 200; c++) {
510 int cols, WhatParamsToTest params_to_test) {
517 RhsType rhs(depth, cols);
520 ResultType result(rows, cols);
539 int cols, WhatParamsToTest params_to_test,
544 MapOrder::ResultOrder>(context, rows, depth, cols, \
761 const OffsetRowDup rhs_offset(b_offset, rhs.cols());
912 const OffsetRowDup rhs_offset(b_offset, rhs.cols());
997 // (result rows) * (result cols) * depth >= 2 x 65 x 1024. Size was selected
1054 const OffsetRowDup rhs_offset(b_offset, rhs.cols());
1202 void TestOutputStages(int rows, int depth, int cols, int result_offset,
1205 Matrix<std::uint8_t, MapOrder::ColMajor> rhs(depth, cols);
1206 Matrix<std::int32_t, ResultOrder> result_raw_int32(rows, cols);
1220 for (int c = 0; c < cols; c++) {
1240 Matrix<std::int32_t, ResultOrder> result_quantized_down_int32(rows, cols);
1247 for (int c = 0; c < cols; c++) {
1255 std::int64_t avg = sum / (rows * cols);
1267 cols);
1274 for (int c = 0; c < cols; c++) {
1282 std::vector<std::int32_t> row_vector_data(cols);
1285 for (int i = 0; i < cols; i++) {
1289 RowVectorMap row_vector_map(row_vector_data.data(), cols);
1293 Matrix<std::int32_t, ResultOrder> result_of_row_bias_addition(rows, cols);
1298 for (int c = 0; c < cols; c++) {
1314 Matrix<std::int32_t, ResultOrder> result_of_col_bias_addition(rows, cols);
1319 for (int c = 0; c < cols; c++) {
1331 for (int c = 0; c < cols; c++) {
1342 Matrix<std::int32_t, ResultOrder> result_clamped(rows, cols);
1347 for (int c = 0; c < cols; c++) {
1362 Matrix<std::int32_t, ResultOrder> result_tanh(rows, cols);
1367 for (int c = 0; c < cols; c++) {
1382 Matrix<std::int32_t, ResultOrder> result_biased_clamped(rows, cols);
1387 for (int c = 0; c < cols; c++) {
1402 rows, cols);
1408 for (int c = 0; c < cols; c++) {
1445 rows, cols);
1453 for (int c = 0; c < cols; c++) {
1473 cols);
1480 for (int c = 0; c < cols; c++) {
