torch_sys::c_generated

Function atg__softmax

source
pub unsafe extern "C" fn atg__softmax(
    out__: *mut *mut C_tensor,
    self_: *mut C_tensor,
    dim_: i64,
    half_to_float_: c_int,
)