Home
last modified time | relevance | path

Searched defs:self_device (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIndexing.h209 const at::Device& self_device, in applySlice()
298 const at::Device& self_device) { in boolToIndexingTensor()
380 const at::Device& self_device) { in scalarToTensor()
519 const at::Device& self_device, in applySlicing()
H A DTensorIndexing.cpp50 at::Device self_device = self.device(); in set_item() local
H A DFunctionalizeFallbackKernel.cpp228 static bool device_opted_into_functionalization(c10::Device self_device, std::optional<c10::Device>… in device_opted_into_functionalization()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_variable_indexing.cpp178 const at::Device& self_device, in applySlicing()
449 at::Device self_device = self_.device(); in THPVariable_setitem() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorAdvancedIndexingUtils.h33 auto self_device = self.device(); in canDispatchToMaskedFill() local