rstsr_core/device_cpu_serial/adv_indexing.rs
1use crate::prelude_dev::*;
2
3impl<T, D> DeviceIndexSelectAPI<T, D> for DeviceCpuSerial
4where
5 T: Clone,
6 D: DimAPI + DimSmallerOneAPI,
7 D::SmallerOne: DimAPI,
8{
9 fn index_select(
10 &self,
11 c: &mut Vec<MaybeUninit<T>>,
12 lc: &Layout<D>,
13 a: &Vec<T>,
14 la: &Layout<D>,
15 axis: usize,
16 indices: &[usize],
17 ) -> Result<()> {
18 index_select_cpu_serial(c, lc, a, la, axis, indices)
19 }
20}