Searched refs:can_fuse_src_batch_dims (Results 1 – 10 of 10) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/matmul/ |
| H A D | gemm_f32_matmul.cpp | 64 .can_fuse_src_batch_dims(); in init()
170 const bool can_fuse_src_batch_dims = pd()->has_runtime_dims_or_strides() in execute_ref() local
171 ? helper.can_fuse_src_batch_dims() in execute_ref()
174 batch, M, N, can_fuse_src_batch_dims); in execute_ref()
184 * ((can_fuse_src_batch_dims || batch == 1) in execute_ref()
203 && (!can_fuse_src_batch_dims || is_binary_po_channel_bcast); in execute_ref()
|
| H A D | gemm_bf16_matmul.cpp | 172 const bool can_fuse_src_batch_dims = pd()->has_runtime_dims_or_strides() in execute_ref() local
173 ? helper.can_fuse_src_batch_dims() in execute_ref()
176 batch, M, N, can_fuse_src_batch_dims); in execute_ref()
186 * ((can_fuse_src_batch_dims || batch == 1) in execute_ref()
207 && (!can_fuse_src_batch_dims || is_binary_po_channel_bcast); in execute_ref()
|
| H A D | gemm_based_common.hpp | 101 const bool can_fuse_src_batch_dims) { in get_scratchpad_size() argument
106 if (can_fuse_src_batch_dims || batch == 1) { in get_scratchpad_size()
|
| H A D | gemm_x8s8s32x_matmul.cpp | 205 const bool can_fuse_src_batch_dims = pd()->has_runtime_dims_or_strides() in execute_ref() local
206 ? helper.can_fuse_src_batch_dims() in execute_ref()
209 batch, M, N, can_fuse_src_batch_dims); in execute_ref()
219 * ((can_fuse_src_batch_dims || batch == 1) in execute_ref()
241 && (!can_fuse_src_batch_dims || is_binary_po_channel_bcast); in execute_ref()
|
| H A D | matmul_utils.hpp | 69 bool can_fuse_src_batch_dims() const { in can_fuse_src_batch_dims() function
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/matmul/ |
| H A D | gemm_f32_matmul.cpp | 68 .can_fuse_src_batch_dims(); in init()
191 const bool can_fuse_src_batch_dims = pd()->has_runtime_dims_or_strides() in execute_ref() local
192 ? helper.can_fuse_src_batch_dims() in execute_ref()
195 batch, M, N, can_fuse_src_batch_dims, nthr); in execute_ref()
205 * ((can_fuse_src_batch_dims || batch == 1) ? 1 : nthr), in execute_ref()
232 const bool parallel_over_batch = batch > 1 && !can_fuse_src_batch_dims; in execute_ref()
|
| H A D | gemm_bf16_matmul.cpp | 187 const bool can_fuse_src_batch_dims = pd()->has_runtime_dims_or_strides() in execute_ref() local
188 ? helper.can_fuse_src_batch_dims() in execute_ref()
191 batch, M, N, can_fuse_src_batch_dims, nthr); in execute_ref()
201 * ((can_fuse_src_batch_dims || batch == 1) ? 1 : nthr), in execute_ref()
230 const bool parallel_over_batch = batch > 1 && !can_fuse_src_batch_dims; in execute_ref()
|
| H A D | gemm_based_common.hpp | 115 const bool can_fuse_src_batch_dims, const int nthr) { in get_scratchpad_size() argument
120 if (can_fuse_src_batch_dims || batch == 1) { in get_scratchpad_size()
|
| H A D | gemm_x8s8s32x_matmul.cpp | 213 const bool can_fuse_src_batch_dims = pd()->has_runtime_dims_or_strides() in execute_ref() local
214 ? helper.can_fuse_src_batch_dims() in execute_ref()
217 batch, M, N, can_fuse_src_batch_dims, nthr); in execute_ref()
227 * ((can_fuse_src_batch_dims || batch == 1) ? 1 : nthr), in execute_ref()
257 const bool parallel_over_batch = batch > 1 && !can_fuse_src_batch_dims; in execute_ref()
|
| H A D | matmul_utils.hpp | 69 bool can_fuse_src_batch_dims() const { in can_fuse_src_batch_dims() function
|