Searched defs:device_indices (Results 1 – 4 of 4) sorted by relevance
41 auto device_indices = fmap(devices_, [](const at::Device& device) -> int64_t { in apply() local
253 let mut device_indices: Vec<usize> = (0..self.device_stats.len()).collect(); in fmt() localVariable
595 std::unordered_set<int> device_indices; in validate_rendezvous_requests() local
19727 pub fn device_indices(mut self, device_indices: &'a [u32]) -> Self { in device_indices() method19883 pub fn device_indices(mut self, device_indices: &'a [u32]) -> Self { in device_indices() method36709 pub fn device_indices(mut self, device_indices: &'a [u32]) -> Self { in device_indices() method