[][src]Function onednn_sys::dnnl_post_ops_get_params_dw_k3s1p1

pub unsafe extern "C" fn dnnl_post_ops_get_params_dw_k3s1p1(
    post_ops: const_dnnl_post_ops_t,
    index: c_int,
    weights_data_type: *mut dnnl_data_type_t,
    bias_data_type: *mut dnnl_data_type_t,
    dst_data_type: *mut dnnl_data_type_t,
    count: *mut dnnl_dim_t,
    mask: *mut c_int,
    scales: *mut *const f32
) -> dnnl_status_t

Returns the parameters of an depthwise post-op with stride 1.

@param post_ops Post-ops. @param index Index of the elementwise post-op. @param weights_data_type Weights data type of depthwise post-op @param bias_data_type Bias data type of depthwise post-op @param dst_data_type Output data type of depthwise post-op @param count Output length of the array of scaling factors @p scales. @param mask Output scaling factors correspondence mask that defines the correspondence between the output tensor dimensions and the @p scales array. The set i-th bit indicates that a dedicated output scaling factor is used for each index along that dimension. The mask value of 0 implies a common scaling factor for the whole output tensor. @param scales Output pointer to a constant array of float scaling factors. @returns #dnnl_success on success and a status describing the error otherwise