Skip to main content

cuda_gather_dispatch

tract_cuda::kernels::array

Function cuda_gather_dispatch 

Source 
pub fn cuda_gather_dispatch(
    data: &DeviceTensor,
    indices: &DeviceTensor,
    axis: usize,
    output: &DeviceTensor,
) -> TractResult<()>