Searched refs:blocksifm (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xsmm_conv2d.cc | 83 int S, int C, int K, int blocksifm, in copy_RSCK_to_custom() argument
87 LIBXSMM_VLA_DECL(6, float, output, kcrs, blocksifm, R, S, ifmblock, ofmblock); in copy_RSCK_to_custom()
91 for (c = 0; c < blocksifm; c++) { in copy_RSCK_to_custom()
97 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
102 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
108 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
219 int blocksifm = in CallLibxsmmConvGeneric() local
224 (float*)libxsmm_aligned_scratch(blocksofm * blocksifm * desc.R * desc.S * in CallLibxsmmConvGeneric()
246 desc.K, blocksifm, blocksofm, ifmblock, ofmblock, in CallLibxsmmConvGeneric()
262 desc.K, blocksifm, blocksofm, ifmblock, ofmblock, in CallLibxsmmConvGeneric()
[all …]
|