pub unsafe extern "C" fn atg_cudnn_batch_norm(
out__: *mut *mut C_tensor,
input_: *mut C_tensor,
weight_: *mut C_tensor,
bias_: *mut C_tensor,
running_mean_: *mut C_tensor,
running_var_: *mut C_tensor,
training_: c_int,
exponential_average_factor_: f64,
epsilon_: f64,
)