Home
last modified time | relevance | path

Searched refs:diff_bia_format (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/
H A Dtest_inner_product_backward_weights.cpp137 memory::format_tag diff_bia_format, in cuda_check_format_tags() argument
159 bool diff_bia_ok = diff_bia_format == memory::format_tag::undef in cuda_check_format_tags()
160 || diff_bia_format == memory::format_tag::any in cuda_check_format_tags()
161 || diff_bia_format == memory::format_tag::a in cuda_check_format_tags()
162 || diff_bia_format == memory::format_tag::x; in cuda_check_format_tags()
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/
H A Dtest_inner_product_backward_weights.cpp137 memory::format_tag diff_bia_format, in cuda_check_format_tags() argument
159 bool diff_bia_ok = diff_bia_format == memory::format_tag::undef in cuda_check_format_tags()
160 || diff_bia_format == memory::format_tag::any in cuda_check_format_tags()
161 || diff_bia_format == memory::format_tag::a in cuda_check_format_tags()
162 || diff_bia_format == memory::format_tag::x; in cuda_check_format_tags()