| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
| H A D | test_inner_product_backward_data.cpp | 86 memory::format_tag diff_dst_format; member
100 p.weights_format, p.diff_dst_format), in SetUp()
108 memory::format_tag wei_format, memory::format_tag diff_dst_format) { in cuda_check_format_tags() argument
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()
166 = create_md({ipd.mb, ipd.oc}, data_type, p.diff_dst_format); in Test()
|
| H A D | test_inner_product_backward_weights.cpp | 113 memory::format_tag diff_dst_format; member
128 p.diff_bias_format, p.diff_dst_format), in SetUp()
138 memory::format_tag diff_dst_format) { in cuda_check_format_tags() argument
163 bool diff_dst_ok = diff_dst_format == memory::format_tag::any in cuda_check_format_tags()
164 || diff_dst_format == memory::format_tag::nc; in cuda_check_format_tags()
203 = create_md({ipd.mb, ipd.oc}, data_type, p.diff_dst_format); in Test()
|
| H A D | test_pooling_backward.cpp | 36 memory::format_tag diff_dst_format; member
284 SKIP_IF_CUDA(!cuda_check_format_tags(p.diff_dst_format), in SetUp()
306 dst_dims, data_type, p.diff_dst_format); in Test()
313 dst_dims, data_type, p.diff_dst_format); in Test()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
| H A D | test_inner_product_backward_data.cpp | 86 memory::format_tag diff_dst_format; member
100 p.weights_format, p.diff_dst_format), in SetUp()
108 memory::format_tag wei_format, memory::format_tag diff_dst_format) { in cuda_check_format_tags() argument
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()
166 = create_md({ipd.mb, ipd.oc}, data_type, p.diff_dst_format); in Test()
|
| H A D | test_inner_product_backward_weights.cpp | 113 memory::format_tag diff_dst_format; member
128 p.diff_bias_format, p.diff_dst_format), in SetUp()
138 memory::format_tag diff_dst_format) { in cuda_check_format_tags() argument
163 bool diff_dst_ok = diff_dst_format == memory::format_tag::any in cuda_check_format_tags()
164 || diff_dst_format == memory::format_tag::nc; in cuda_check_format_tags()
203 = create_md({ipd.mb, ipd.oc}, data_type, p.diff_dst_format); in Test()
|
| H A D | test_pooling_backward.cpp | 36 memory::format_tag diff_dst_format; member
284 SKIP_IF_CUDA(!cuda_check_format_tags(p.diff_dst_format), in SetUp()
306 dst_dims, data_type, p.diff_dst_format); in Test()
313 dst_dims, data_type, p.diff_dst_format); in Test()
|
| /dports/math/ideep/ideep-2.0.0-119-gb57539e/tests/ |
| H A D | test_ideep_inner_product_backward_data.cc | 21 static_cast<format>(p.diff_dst_format)); in SetUp()
|
| H A D | test_ideep_inner_product_backward_weights.cc | 31 static_cast<format>(p.diff_dst_format)); in SetUp()
|
| H A D | test_ideep_common.hpp | 560 mkldnn::memory::format diff_dst_format; member
1070 mkldnn::memory::format diff_dst_format; member
1081 mkldnn::memory::format diff_dst_format; member
|
| H A D | test_ideep_pooling_backward.cc | 25 data_type, static_cast<format>(p.diff_dst_format)); in TestCommon()
|