torch_sys

Function at_loadz_callback_with_device

source
pub unsafe extern "C" fn at_loadz_callback_with_device(
    filename: *const c_char,
    data: *mut c_void,
    f: extern "C" fn(_: *mut c_void, name: *const c_char, t: *mut C_tensor),
    device_id: c_int,
)