Searched refs:blocksifm (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xsmm_conv2d.cc | 86 int S, int C, int K, int blocksifm, in copy_RSCK_to_custom() argument
90 LIBXSMM_VLA_DECL(6, float, output, kcrs, blocksifm, R, S, ifmblock, ofmblock); in copy_RSCK_to_custom()
94 for (c = 0; c < blocksifm; c++) { in copy_RSCK_to_custom()
100 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
105 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
111 v2 - k * ofmblock, blocksifm, R, S, ifmblock, in copy_RSCK_to_custom()
275 const int blocksifm = in CallLibxsmmConvGeneric() local
281 blocksofm * blocksifm * desc.R * desc.S * ifmblock * ofmblock; in CallLibxsmmConvGeneric()
302 desc.K, blocksifm, blocksofm, ifmblock, ofmblock, in CallLibxsmmConvGeneric()
318 desc.K, blocksifm, blocksofm, ifmblock, ofmblock, in CallLibxsmmConvGeneric()
|