Searched refs:blocksofm (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
221 int blocksofm = in CallLibxsmmConvGeneric() local
224 (float*)libxsmm_aligned_scratch(blocksofm * blocksifm * desc.R * desc.S * in CallLibxsmmConvGeneric()
236 if (blocksofm > num_threads) { in CallLibxsmmConvGeneric()
237 int work = blocksofm; in CallLibxsmmConvGeneric()
246 desc.K, blocksifm, blocksofm, ifmblock, ofmblock, in CallLibxsmmConvGeneric()
253 int work = blocksofm; in CallLibxsmmConvGeneric()
262 desc.K, blocksifm, blocksofm, ifmblock, ofmblock, in CallLibxsmmConvGeneric()
281 blocksofm * blocksifm * desc.R * desc.S * ifmblock * ofmblock * in CallLibxsmmConvGeneric()
|