ml.h

Go to the documentation of this file.

/*******************************************************
 * Copyright (c) 2018, ArrayFire
 * All rights reserved.
 *
 * This file is distributed under 3-clause BSD license.
 * The complete license agreement can be obtained at:
 * http://arrayfire.com/licenses/BSD-3-Clause
 ********************************************************/

#pragma once
#include <af/defines.h>

#ifdef __cplusplus
namespace af
{
class array;
class dim4;

#if AF_API_VERSION >= 37

    AFAPI array convolve2GradientNN(const array& incoming_gradient,
                                    const array& original_signal,
                                    const array& original_filter,
                                    const array& convolved_output,
                                    const dim4 stride, const dim4 padding, const dim4 dilation,
                                    convGradientType grad_type);

#endif

}
#endif

#ifdef __cplusplus
extern "C" {
#endif

#if AF_API_VERSION >= 37

    AFAPI af_err af_convolve2_gradient_nn(af_array *out,
                                          const af_array incoming_gradient,
                                          const af_array original_signal,
                                          const af_array original_filter,
                                          const af_array convolved_output,
                                          const unsigned stride_dims,   const dim_t *strides,
                                          const unsigned padding_dims,  const dim_t *paddings,
                                          const unsigned dilation_dims, const dim_t *dilations,
                                          af_conv_gradient_type grad_type);
#endif


#ifdef __cplusplus
}
#endif