pub unsafe extern "C" fn atg_softmax(
    out__: *mut *mut C_tensor,
    self_: *mut C_tensor,
    dim_: i64,
    dtype_: c_int
)