Searched refs:blocksofm (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xsmm_conv2d.cc | 87 int blocksofm, int ifmblock, in copy_RSCK_to_custom() argument
277 const int blocksofm = in CallLibxsmmConvGeneric() local
281 blocksofm * blocksifm * desc.R * desc.S * ifmblock * ofmblock; in CallLibxsmmConvGeneric()
292 if (blocksofm > num_threads) { in CallLibxsmmConvGeneric()
293 const int work = blocksofm; in CallLibxsmmConvGeneric()
302 desc.K, blocksifm, blocksofm, ifmblock, ofmblock, in CallLibxsmmConvGeneric()
309 const int work = blocksofm; in CallLibxsmmConvGeneric()
318 desc.K, blocksifm, blocksofm, ifmblock, ofmblock, in CallLibxsmmConvGeneric()
|