| D | xsmm_conv2d_test.cc | 45 int kh; member
129 int kh = param->kh; in naive_conv_fp() local
140 LIBXSMM_VLA_DECL(4, const float, filter_t, filter, nIfm, kh, kw); in naive_conv_fp()
149 for (kj = 0; kj < kh; ++kj) { in naive_conv_fp()
157 LIBXSMM_VLA_ACCESS(4, filter_t, ofm, ifm, kj, ki, nIfm, kh, in naive_conv_fp()
195 int kh = 3; /* filter height, "R" */ in TEST_F() local
212 int ofh = (ifh + 2 * pad_h - kh) / stride_h + 1; in TEST_F()
227 nOfm * nIfm * kh * kw * sizeof(float), 2097152); in TEST_F()
231 init_buf(naive_filter, nOfm * nIfm * kh * kw, 0, 0); in TEST_F()
235 Tensor filter(DT_FLOAT, {kh, kw, nIfm, nOfm}); in TEST_F()
[all …]
|