Home
last modified time | relevance | path

Searched defs:input2_device (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/
H A DBlas.cpp402 auto input2_device = input2.device(); in tensordot_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLinear.cpp833 auto input2_device = input2.device(); in tensordot_out() local
H A DLinearAlgebra.cpp1896 auto input2_device = other.device(); in dot_out() local
1912 auto input2_device = other.device(); in vdot_out() local