torch_sys::c_generated

Function atg_embedding

source
pub unsafe extern "C" fn atg_embedding(
    out__: *mut *mut C_tensor,
    weight_: *mut C_tensor,
    indices_: *mut C_tensor,
    padding_idx_: i64,
    scale_grad_by_freq_: c_int,
    sparse_: c_int,
)