Lines Matching refs:ks_index
345 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
347 im2col[ks_index * mr() + m_index] = a.data() + a_stride() * m_index - a_offset();
353 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
354 im2col[ks_index * mr() + zero_index()] = a.data();
357 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
359 im2col[ks_index * mr() + m_index] = junk.data();
367 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
369 if (im2col[ks_index * mr() + m_index] == a.data()) {
371 (int32_t(im2col[ks_index * mr() + m_index][k_index]) - int32_t(a_zero_point())) *
372 … (int32_t(b[(n_index * ks() + ks_index) * k() + k_index]) - int32_t(b_zero_point()));
375 … (int32_t(im2col[ks_index * mr() + m_index][k_index + a_offset()]) - int32_t(a_zero_point())) *
376 … (int32_t(b[(n_index * ks() + ks_index) * k() + k_index]) - int32_t(b_zero_point()));
659 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
661 im2col[ks_index * mr() + m_index] = a.data() + a_stride() * m_index - a_offset();
667 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
668 im2col[ks_index * mr() + zero_index()] = a.data();
671 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
673 im2col[ks_index * mr() + m_index] = junk.data();
681 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
683 if (im2col[ks_index * mr() + m_index] == a.data()) {
685 … (int32_t(im2col[ks_index * mr() + m_index][k_index]) - int32_t(a_zero_point() - 0x80)) *
686 int32_t(b[(n_index * ks() + ks_index) * k() + k_index]);
689 …(int32_t(im2col[ks_index * mr() + m_index][k_index + a_offset()]) - int32_t(a_zero_point() - 0x80)…
690 int32_t(b[(n_index * ks() + ks_index) * k() + k_index]);
853 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
855 im2col[ks_index * mr() + m_index] = a.data() + a_stride() * m_index - a_offset(); in Test()
860 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
861 im2col[ks_index * mr() + zero_index()] = a.data(); in Test()
864 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
866 im2col[ks_index * mr() + m_index] = junk.data(); in Test()
873 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
875 ASSERT_LT(ks_index * mr() + m_index, im2col.size()); in Test()
878 if (im2col[ks_index * mr() + m_index] == a.data()) { in Test()
880 fp16_ieee_to_fp32_value(im2col[ks_index * mr() + m_index][k_index]) * in Test()
881 fp16_ieee_to_fp32_value(b[(n_index * ks() + ks_index) * k() + k_index]); in Test()
884 fp16_ieee_to_fp32_value(im2col[ks_index * mr() + m_index][k_index + a_offset()]) * in Test()
885 fp16_ieee_to_fp32_value(b[(n_index * ks() + ks_index) * k() + k_index]); in Test()
1359 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
1361 im2col[ks_index * mr() + m_index] = a.data() + a_stride() * m_index - a_offset(); in Test()
1366 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
1367 im2col[ks_index * mr() + zero_index()] = a.data(); in Test()
1370 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
1372 im2col[ks_index * mr() + m_index] = junk.data(); in Test()
1379 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
1381 ASSERT_LT(ks_index * mr() + m_index, im2col.size()); in Test()
1384 if (im2col[ks_index * mr() + m_index] == a.data()) { in Test()
1386 (im2col[ks_index * mr() + m_index][k_index]) * in Test()
1387 (b[(n_index * ks() + ks_index) * k() + k_index]); in Test()
1390 (im2col[ks_index * mr() + m_index][k_index + a_offset()]) * in Test()
1391 (b[(n_index * ks() + ks_index) * k() + k_index]); in Test()
1451 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
1453 im2col[ks_index * mr() + m_index] = a.data() + a_stride() * m_index - a_offset(); in Test()
1458 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
1459 im2col[ks_index * mr() + zero_index()] = a.data(); in Test()
1462 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
1464 im2col[ks_index * mr() + m_index] = junk.data(); in Test()
1471 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { in Test() local
1473 ASSERT_LT(ks_index * mr() + m_index, im2col.size()); in Test()
1476 if (im2col[ks_index * mr() + m_index] == a.data()) { in Test()
1478 (im2col[ks_index * mr() + m_index][k_index]) * in Test()
1479 (b[(n_index * ks() + ks_index) * k() + k_index]); in Test()
1482 (im2col[ks_index * mr() + m_index][k_index + a_offset()]) * in Test()
1483 (b[(n_index * ks() + ks_index) * k() + k_index]); in Test()
1547 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
1549 im2col[ks_index * mr() + m_index] = a.data() + a_stride() * m_index - a_offset();
1554 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
1555 im2col[ks_index * mr() + zero_index()] = a.data();
1558 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
1560 im2col[ks_index * mr() + m_index] = junk.data();
1567 for (size_t ks_index = 0; ks_index < ks(); ks_index++) { variable
1569 ASSERT_LT(ks_index * mr() + m_index, im2col.size());
1572 if (im2col[ks_index * mr() + m_index] == a.data()) {
1574 (im2col[ks_index * mr() + m_index][k_index]) *
1575 (b[(n_index * ks() + ks_index) * k() + k_index]);
1578 (im2col[ks_index * mr() + m_index][k_index + a_offset()]) *
1579 (b[(n_index * ks() + ks_index) * k() + k_index]);