Searched refs:ifmblock (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xsmm_conv2d.cc | 84 int blocksofm, int ifmblock, in copy_RSCK_to_custom() argument
87 LIBXSMM_VLA_DECL(6, float, output, kcrs, blocksifm, R, S, ifmblock, ofmblock); in copy_RSCK_to_custom()
94 for (v1 = c * ifmblock; v1 < std::min(C, (c + 1) * ifmblock); v1++) { in copy_RSCK_to_custom()
96 LIBXSMM_VLA_ACCESS(6, output, k, c, r, s, v1 - c * ifmblock, in copy_RSCK_to_custom()
97 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
101 LIBXSMM_VLA_ACCESS(6, output, k, c, r, s, v1 - c * ifmblock, in copy_RSCK_to_custom()
102 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
105 for (v1 = C; v1 < (c + 1) * ifmblock; v1++) { in copy_RSCK_to_custom()
107 LIBXSMM_VLA_ACCESS(6, output, k, c, r, s, v1 - c * ifmblock, in copy_RSCK_to_custom()
108 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
[all …]
|