Skip to main content

compute_broadcast_strides

tract_gpu::utils

Function compute_broadcast_strides 

Source 
pub fn compute_broadcast_strides<T: Zero + Copy + 'static>(
    shape: &[usize],
    strides: &[isize],
) -> TractResult<TVec<T>>
where
    isize: AsPrimitive<T>,