Lines Matching refs:memory
25 memory::dim mb;
26 memory::dim ic;
27 memory::dim oc;
28 memory::dim kd, kh, kw;
33 const test_inner_product_descr_t &ipd, const memory &diff_dst, in compute_ref_inner_product_bwd_data()
34 const memory &weights, const memory &diff_src) { in compute_ref_inner_product_bwd_data()
39 const memory::desc diff_dst_d = diff_dst.get_desc(); in compute_ref_inner_product_bwd_data()
40 const memory::desc weights_d = weights.get_desc(); in compute_ref_inner_product_bwd_data()
41 const memory::desc diff_src_d = diff_src.get_desc(); in compute_ref_inner_product_bwd_data()
50 dnnl::impl::parallel_nd(ipd.mb, ipd.ic, [&](memory::dim n, memory::dim ic) { in compute_ref_inner_product_bwd_data()
52 for_(memory::dim kd = 0; kd < ipd.kd; ++kd) in compute_ref_inner_product_bwd_data()
53 for_(memory::dim kh = 0; kh < ipd.kh; ++kh) in compute_ref_inner_product_bwd_data()
54 for (memory::dim kw = 0; kw < ipd.kw; ++kw) { in compute_ref_inner_product_bwd_data()
55 memory::dim dsidx = n * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_data()
60 for (memory::dim oc = 0; oc < ipd.oc; ++oc) { in compute_ref_inner_product_bwd_data()
61 memory::dim ddidx = n * ipd.oc + oc; in compute_ref_inner_product_bwd_data()
62 memory::dim widx = oc * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_data()
70 memory::dim dsidx = n * ipd.ic + ic; in compute_ref_inner_product_bwd_data()
73 for (memory::dim oc = 0; oc < ipd.oc; ++oc) { in compute_ref_inner_product_bwd_data()
74 memory::dim ddidx = n * ipd.oc + oc; in compute_ref_inner_product_bwd_data()
75 memory::dim widx = oc * ipd.ic + ic; in compute_ref_inner_product_bwd_data()
84 memory::format_tag diff_src_format;
85 memory::format_tag weights_format;
86 memory::format_tag diff_dst_format;
107 bool cuda_check_format_tags(memory::format_tag diff_src_format, in cuda_check_format_tags()
108 memory::format_tag wei_format, memory::format_tag diff_dst_format) { in cuda_check_format_tags()
109 bool diff_src_ok = diff_src_format == memory::format_tag::ncdhw in cuda_check_format_tags()
110 || diff_src_format == memory::format_tag::ndhwc in cuda_check_format_tags()
111 || diff_src_format == memory::format_tag::nchw in cuda_check_format_tags()
112 || diff_src_format == memory::format_tag::nhwc in cuda_check_format_tags()
113 || diff_src_format == memory::format_tag::ncw in cuda_check_format_tags()
114 || diff_src_format == memory::format_tag::nwc in cuda_check_format_tags()
115 || diff_src_format == memory::format_tag::nc in cuda_check_format_tags()
116 || diff_src_format == memory::format_tag::any; in cuda_check_format_tags()
117 bool wei_ok = wei_format == memory::format_tag::oidhw in cuda_check_format_tags()
118 || wei_format == memory::format_tag::odhwi in cuda_check_format_tags()
119 || wei_format == memory::format_tag::dhwio in cuda_check_format_tags()
120 || wei_format == memory::format_tag::oihw in cuda_check_format_tags()
121 || wei_format == memory::format_tag::hwio in cuda_check_format_tags()
122 || wei_format == memory::format_tag::ohwi in cuda_check_format_tags()
123 || wei_format == memory::format_tag::oiw in cuda_check_format_tags()
124 || wei_format == memory::format_tag::owi in cuda_check_format_tags()
125 || wei_format == memory::format_tag::wio in cuda_check_format_tags()
126 || wei_format == memory::format_tag::io in cuda_check_format_tags()
127 || wei_format == memory::format_tag::oi in cuda_check_format_tags()
128 || wei_format == memory::format_tag::any; in cuda_check_format_tags()
129 bool diff_dst_ok = diff_dst_format == memory::format_tag::any in cuda_check_format_tags()
130 || diff_dst_format == memory::format_tag::nc; in cuda_check_format_tags()
143 memory::data_type data_type = data_traits<data_t>::data_type; in Test()
144 ASSERT_EQ(data_type, dnnl::memory::data_type::f32); in Test()
146 memory::dims diff_src_dims = {ipd.mb, ipd.ic}, in Test()
239 ::testing::Values(inprod_test_params_float {memory::format_tag::any,
240 memory::format_tag::any, memory::format_tag::any,
246 inprod_test_params_float {memory::format_tag::any,
247 memory::format_tag::any, memory::format_tag::any,
250 inprod_test_params_float {memory::format_tag::any,
251 memory::format_tag::any, memory::format_tag::any,
254 inprod_test_params_float {memory::format_tag::any,
255 memory::format_tag::any, memory::format_tag::any,
262 inprod_test_params_float {memory::format_tag::nCdhw8c,
263 memory::format_tag::aBcde8b, memory::format_tag::nc,
265 inprod_test_params_float {memory::format_tag::nCdhw8c,
266 memory::format_tag::aBcde8b, memory::format_tag::nc,
268 inprod_test_params_float {memory::format_tag::nCdhw8c,
269 memory::format_tag::aBcde8b, memory::format_tag::nc,
275 inprod_test_params_float {memory::format_tag::nCdhw16c,
276 memory::format_tag::aBcde16b, memory::format_tag::nc,
278 inprod_test_params_float {memory::format_tag::nCdhw16c,
279 memory::format_tag::aBcde16b, memory::format_tag::nc,
281 inprod_test_params_float {memory::format_tag::nCdhw16c,
282 memory::format_tag::aBcde16b, memory::format_tag::nc,
288 inprod_test_params_float {memory::format_tag::nChw16c,
289 memory::format_tag::aBcd16b, memory::format_tag::nc,
291 inprod_test_params_float {memory::format_tag::nChw16c,
292 memory::format_tag::aBcd16b, memory::format_tag::nc,
294 inprod_test_params_float {memory::format_tag::nChw16c,
295 memory::format_tag::aBcd16b, memory::format_tag::nc,
297 inprod_test_params_float {memory::format_tag::nChw8c,
298 memory::format_tag::aBcd8b, memory::format_tag::nc,
300 inprod_test_params_float {memory::format_tag::nChw8c,
301 memory::format_tag::aBcd8b, memory::format_tag::nc,
303 inprod_test_params_float {memory::format_tag::nChw8c,
304 memory::format_tag::aBcd8b, memory::format_tag::nc,
309 inprod_test_params_float {memory::format_tag::any,
310 memory::format_tag::any, memory::format_tag::any,
312 inprod_test_params_float {memory::format_tag::any,
313 memory::format_tag::any, memory::format_tag::any,
315 inprod_test_params_float {memory::format_tag::nwc,
316 memory::format_tag::oiw, memory::format_tag::nc,
318 inprod_test_params_float {memory::format_tag::nwc,
319 memory::format_tag::wio, memory::format_tag::nc,
321 inprod_test_params_float {memory::format_tag::nwc,
322 memory::format_tag::owi, memory::format_tag::nc,
324 inprod_test_params_float {memory::format_tag::ncw,
325 memory::format_tag::oiw, memory::format_tag::nc,
327 inprod_test_params_float {memory::format_tag::ncw,
328 memory::format_tag::wio, memory::format_tag::nc,
330 inprod_test_params_float {memory::format_tag::ncw,
331 memory::format_tag::owi, memory::format_tag::nc,
333 inprod_test_params_float {memory::format_tag::nhwc,
334 memory::format_tag::hwio, memory::format_tag::nc,
336 inprod_test_params_float {memory::format_tag::nhwc,
337 memory::format_tag::oihw, memory::format_tag::nc,
339 inprod_test_params_float {memory::format_tag::nhwc,
340 memory::format_tag::iohw, memory::format_tag::nc,
342 inprod_test_params_float {memory::format_tag::nchw,
343 memory::format_tag::oihw, memory::format_tag::nc,
345 inprod_test_params_float {memory::format_tag::nchw,
346 memory::format_tag::hwio, memory::format_tag::nc,
348 inprod_test_params_float {memory::format_tag::nchw,
349 memory::format_tag::ohwi, memory::format_tag::nc,
351 inprod_test_params_float {memory::format_tag::nChw8c,
352 memory::format_tag::aBcd8b, memory::format_tag::nc,
354 inprod_test_params_float {memory::format_tag::any,
355 memory::format_tag::aBcd8b, memory::format_tag::nc,
357 inprod_test_params_float {memory::format_tag::nChw8c,
358 memory::format_tag::any, memory::format_tag::nc,
360 inprod_test_params_float {memory::format_tag::nChw8c,
361 memory::format_tag::aBcd8b, memory::format_tag::nc,
363 inprod_test_params_float {memory::format_tag::nChw16c,
364 memory::format_tag::aBcd16b, memory::format_tag::nc,
366 inprod_test_params_float {memory::format_tag::nc,
367 memory::format_tag::oi, memory::format_tag::nc,
369 inprod_test_params_float {memory::format_tag::nc,
370 memory::format_tag::oi, memory::format_tag::nc,
372 inprod_test_params_float {memory::format_tag::nc,
373 memory::format_tag::io, memory::format_tag::nc,
379 inprod_test_params_float {memory::format_tag::any,
380 memory::format_tag::any, memory::format_tag::any,
382 inprod_test_params_float {memory::format_tag::any,
383 memory::format_tag::any, memory::format_tag::any,
385 inprod_test_params_float {memory::format_tag::ncdhw,
386 memory::format_tag::oidhw, memory::format_tag::nc,
388 inprod_test_params_float {memory::format_tag::ncdhw,
389 memory::format_tag::odhwi, memory::format_tag::nc,
391 inprod_test_params_float {memory::format_tag::ncdhw,
392 memory::format_tag::dhwio, memory::format_tag::nc,
394 inprod_test_params_float {memory::format_tag::ndhwc,
395 memory::format_tag::oidhw, memory::format_tag::nc,
397 inprod_test_params_float {memory::format_tag::ndhwc,
398 memory::format_tag::odhwi, memory::format_tag::nc,
400 inprod_test_params_float {memory::format_tag::ndhwc,
401 memory::format_tag::dhwio, memory::format_tag::nc,