Searched refs:input1_device (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Blas.cpp | 401 auto input1_device = input1.device(); in tensordot_out() local
405 (output_device == input1_device) && (input1_device == input2_device), in tensordot_out()
410 input1_device, in tensordot_out()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Linear.cpp | 832 auto input1_device = input1.device(); in tensordot_out() local
836 (output_device == input1_device) && (input1_device == input2_device), in tensordot_out()
839 ", input tensor a on ", input1_device, ", and input tensor b on ", input2_device); in tensordot_out()
|
| H A D | LinearAlgebra.cpp | 1895 auto input1_device = self.device(); in dot_out() local
1899 (output_device == input1_device) && (input1_device == input2_device), in dot_out()
1902 ", the 'input' tensor on ", input1_device, ", and the 'other' tensor on ", input2_device); in dot_out()
1911 auto input1_device = self.device(); in vdot_out() local
1915 (output_device == input1_device) && (input1_device == input2_device), in vdot_out()
1918 ", the 'input' tensor on ", input1_device, ", and the 'other' tensor on ", input2_device); in vdot_out()
|