torch_sys::c_generated

Function atg_cudnn_convolution_add_relu

source
pub unsafe extern "C" fn atg_cudnn_convolution_add_relu(
    out__: *mut *mut C_tensor,
    self_: *mut C_tensor,
    weight_: *mut C_tensor,
    z_: *mut C_tensor,
    alpha_: *mut C_scalar,
    bias_: *mut C_tensor,
    stride_data: *const i64,
    stride_len: c_int,
    padding_data: *const i64,
    padding_len: c_int,
    dilation_data: *const i64,
    dilation_len: c_int,
    groups_: i64,
)