Searched refs:nImg (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xsmm_conv2d_test.cc | 28 int nImg; member
112 int nImg = param->nImg; in naive_conv_fp() local
142 for (img = 0; img < nImg; ++img) { in naive_conv_fp()
192 int nImg = 32; /* mini-batch size, "N" */ in TEST_F() local
223 nImg * nIfm * ifhp * ifwp * sizeof(float), 2097152); in TEST_F()
225 nImg * nOfm * ofhp * ofwp * sizeof(float), 2097152); in TEST_F()
229 init_buf(naive_input, nImg * nIfm * ifhp * ifwp, 0, 0); in TEST_F()
230 zero_buf(naive_output, nImg * nOfm * ofhp * ofwp); in TEST_F()
233 Tensor image(DT_FLOAT, {nImg, ifhp, ifwp, nIfm}); in TEST_F()
237 naive_copy_NCHW_to_NHWC(naive_input, image, nImg, ifhp, ifwp, nIfm); in TEST_F()
[all …]
|