Searched refs:scratchpad_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| H A D | Deconv.cpp | 227 at::Tensor scratchpad_tensor = at::empty( in deconvolution() local
230 deconv_fwd_pd.scratchpad_desc(), engine, scratchpad_tensor.data_ptr()); in deconvolution()
313 at::Tensor scratchpad_tensor = at::empty( in deconvolution_backward_data() local
318 scratchpad_tensor.data_ptr()); in deconvolution_backward_data()
417 at::Tensor scratchpad_tensor = at::empty( in deconvolution_backward_weights() local
420 deconv_bwd_w_pd.scratchpad_desc(), engine, scratchpad_tensor.data_ptr()); in deconvolution_backward_weights()
|
| H A D | Conv.cpp | 240 at::Tensor scratchpad_tensor = at::empty( in convolution() local
243 conv_fwd_pd.scratchpad_desc(), engine, scratchpad_tensor.data_ptr()); in convolution()
341 at::Tensor scratchpad_tensor = at::empty( in convolution_backward_weights() local
344 conv_bwd_w_pd.scratchpad_desc(), engine, scratchpad_tensor.data_ptr()); in convolution_backward_weights()
432 at::Tensor scratchpad_tensor = at::empty( in convolution_backward_data() local
437 scratchpad_tensor.data_ptr()); in convolution_backward_data()
|
| H A D | Matmul.cpp | 222 at::Tensor scratchpad_tensor = at::empty( in matmul() local
225 matmul_pd.scratchpad_desc(), engine, scratchpad_tensor.data_ptr()); in matmul()
|