Skip to main content

copy_gpu_ref

Function copy_gpu_ref 

Source
pub fn copy_gpu_ref<R: Runtime>(
    client: &ComputeClient<R>,
    input: &TensorHandleRef<'_, R>,
    output: &TensorHandleRef<'_, R>,
    dtype: StorageType,
) -> Result<(), LaunchError>
Expand description

Make a jit tensor contiguous.