Searched refs:mem_wrapper (Results 1 – 8 of 8) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/nvidia/ |
| H A D | cudnn_softmax_impl.hpp | 139 const memory_desc_wrapper mem_wrapper(md); in convert_tag() local
140 if (mem_wrapper.matches_one_of_tag(format_tag::ba)) { in convert_tag()
142 } else if (mem_wrapper.matches_one_of_tag(format_tag::ab, in convert_tag()
146 } else if (mem_wrapper.matches_one_of_tag(format_tag::acb, in convert_tag()
|
| H A D | cudnn_conv_filter_adjustment_base.hpp | 41 const memory_desc_wrapper mem_wrapper(md); in supported_filter_format() local
46 return (!(mem_wrapper.matches_one_of_tag(/*format_tag::oidhw,*/ in supported_filter_format()
|
| H A D | sycl_cuda_utils.hpp | 133 const memory_desc_wrapper mem_wrapper(md); in get_format() local
136 } else if (mem_wrapper.matches_one_of_tag(format_tag::ab, format_tag::abc, in get_format()
140 } else if (mem_wrapper.matches_one_of_tag( in get_format()
146 if (consider_ab_as_nhwc && mem_wrapper.matches_one_of_tag(format_tag::ab)) { in get_format()
|
| H A D | cudnn_convolution_impl.hpp | 77 const memory_desc_wrapper mem_wrapper(md); in supported_filter_format() local
79 return (mem_wrapper.matches_one_of_tag(format_tag::ab, format_tag::abc, in supported_filter_format()
81 || (with_groups ? mem_wrapper.matches_one_of_tag( in supported_filter_format()
84 : mem_wrapper.matches_one_of_tag( in supported_filter_format()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/nvidia/ |
| H A D | cudnn_softmax_impl.hpp | 139 const memory_desc_wrapper mem_wrapper(md); in convert_tag() local
140 if (mem_wrapper.matches_one_of_tag(format_tag::ba)) { in convert_tag()
142 } else if (mem_wrapper.matches_one_of_tag(format_tag::ab, in convert_tag()
146 } else if (mem_wrapper.matches_one_of_tag(format_tag::acb, in convert_tag()
|
| H A D | cudnn_conv_filter_adjustment_base.hpp | 41 const memory_desc_wrapper mem_wrapper(md); in supported_filter_format() local
46 return (!(mem_wrapper.matches_one_of_tag(/*format_tag::oidhw,*/ in supported_filter_format()
|
| H A D | sycl_cuda_utils.hpp | 134 const memory_desc_wrapper mem_wrapper(md); in get_format() local
137 } else if (mem_wrapper.matches_one_of_tag(format_tag::ab, format_tag::abc, in get_format()
141 } else if (mem_wrapper.matches_one_of_tag( in get_format()
147 if (consider_ab_as_nhwc && mem_wrapper.matches_one_of_tag(format_tag::ab)) { in get_format()
|
| H A D | cudnn_convolution_impl.hpp | 80 const memory_desc_wrapper mem_wrapper(md); in supported_filter_format() local
82 return (mem_wrapper.matches_one_of_tag(format_tag::ab, format_tag::abc, in supported_filter_format()
84 || (with_groups ? mem_wrapper.matches_one_of_tag( in supported_filter_format()
87 : mem_wrapper.matches_one_of_tag( in supported_filter_format()
|