Searched refs:ofmblock (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xsmm_conv2d.cc | 85 int ofmblock, int start, int end) { in copy_RSCK_to_custom() argument
87 LIBXSMM_VLA_DECL(6, float, output, kcrs, blocksifm, R, S, ifmblock, ofmblock); in copy_RSCK_to_custom()
95 for (v2 = k * ofmblock; v2 < std::min(K, (k + 1) * ofmblock); v2++) in copy_RSCK_to_custom()
97 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
98 ofmblock) = in copy_RSCK_to_custom()
100 for (v2 = K; v2 < (k + 1) * ofmblock; v2++) in copy_RSCK_to_custom()
102 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
103 ofmblock) = 0.0f; in copy_RSCK_to_custom()
106 for (v2 = k * ofmblock; v2 < (k + 1) * ofmblock; v2++) in copy_RSCK_to_custom()
108 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
[all …]
|