Searched refs:nOfm (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xsmm_conv2d_test.cc | 30 int nOfm; member
114 int nOfm = param->nOfm; in naive_conv_fp() local
137 nOfm, ofhp, ofwp); in naive_conv_fp()
143 for (ofm = 0; ofm < nOfm; ++ofm) { in naive_conv_fp()
153 LIBXSMM_VLA_ACCESS(4, output_t, img, ofm, oj, oi, nOfm, ofhp, in naive_conv_fp()
194 int nOfm = 64; /* number of output feature maps, "K" */ in TEST_F() local
225 nImg * nOfm * ofhp * ofwp * sizeof(float), 2097152); in TEST_F()
227 nOfm * nIfm * kh * kw * sizeof(float), 2097152); in TEST_F()
230 zero_buf(naive_output, nImg * nOfm * ofhp * ofwp); in TEST_F()
231 init_buf(naive_filter, nOfm * nIfm * kh * kw, 0, 0); in TEST_F()
[all …]
|