Searched refs:scratchpad_size (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| D | Conv.cpp | 239 size_t scratchpad_size = conv_fwd_pd.scratchpad_desc().get_size(); in convolution() local
241 {static_cast<int64_t>(scratchpad_size)}, src.options().dtype(at::kByte), std::nullopt); in convolution()
340 size_t scratchpad_size = conv_bwd_w_pd.scratchpad_desc().get_size(); in convolution_backward_weights() local
342 {static_cast<int64_t>(scratchpad_size)}, src.options().dtype(at::kByte), std::nullopt); in convolution_backward_weights()
431 size_t scratchpad_size = conv_backward_data_pd.scratchpad_desc().get_size(); in convolution_backward_data() local
433 {static_cast<int64_t>(scratchpad_size)}, diff_dst.options().dtype(at::kByte), std::nullopt); in convolution_backward_data()
|
| D | Deconv.cpp | 226 size_t scratchpad_size = deconv_fwd_pd.scratchpad_desc().get_size(); in deconvolution() local
228 {static_cast<int64_t>(scratchpad_size)}, src.options().dtype(at::kByte), std::nullopt); in deconvolution()
312 size_t scratchpad_size = deconv_backward_data_pd.scratchpad_desc().get_size(); in deconvolution_backward_data() local
314 {static_cast<int64_t>(scratchpad_size)}, diff_dst.options().dtype(at::kByte), std::nullopt); in deconvolution_backward_data()
416 size_t scratchpad_size = deconv_bwd_w_pd.scratchpad_desc().get_size(); in deconvolution_backward_weights() local
418 {static_cast<int64_t>(scratchpad_size)}, src.options().dtype(at::kByte), std::nullopt); in deconvolution_backward_weights()
|
| D | Matmul.cpp | 221 size_t scratchpad_size = matmul_pd.scratchpad_desc().get_size(); in matmul() local
223 {static_cast<int64_t>(scratchpad_size)}, m1.options().dtype(at::kByte), std::nullopt); in matmul()
|