ml.h
Go to the documentation of this file.
1 /*******************************************************
2  * Copyright (c) 2018, ArrayFire
3  * All rights reserved.
4  *
5  * This file is distributed under 3-clause BSD license.
6  * The complete license agreement can be obtained at:
7  * http://arrayfire.com/licenses/BSD-3-Clause
8  ********************************************************/
9 
10 #pragma once
11 #include <af/defines.h>
12 
13 #ifdef __cplusplus
14 namespace af
15 {
16 class array;
17 class dim4;
18 
19 #if AF_API_VERSION >= 37
20 
40  AFAPI array convolve2GradientNN(const array& incoming_gradient,
41  const array& original_signal,
42  const array& original_filter,
43  const array& convolved_output,
44  const dim4 stride, const dim4 padding, const dim4 dilation,
45  convGradientType grad_type);
46 
47 #endif
48 
49 }
50 #endif
51 
52 #ifdef __cplusplus
53 extern "C" {
54 #endif
55 
56 #if AF_API_VERSION >= 37
57 
82  const af_array incoming_gradient,
83  const af_array original_signal,
84  const af_array original_filter,
85  const af_array convolved_output,
86  const unsigned stride_dims, const dim_t *strides,
87  const unsigned padding_dims, const dim_t *paddings,
88  const unsigned dilation_dims, const dim_t *dilations,
89  af_conv_gradient_type grad_type);
90 #endif
91 
92 
93 #ifdef __cplusplus
94 }
95 #endif
af_conv_gradient_type
af_conv_gradient_type
Definition: defines.h:538
dim_t
long long dim_t
Definition: defines.h:56
AFAPI
#define AFAPI
Definition: defines.h:38
af
Definition: algorithm.h:14
af::convGradientType
af_conv_gradient_type convGradientType
Definition: defines.h:599
af_array
void * af_array
Definition: defines.h:240
afcl::array
static af::array array(af::dim4 idims, cl_mem buf, af::dtype type, bool retain=false)
Create an af::array object from an OpenCL cl_mem buffer.
Definition: opencl.h:327
af::convolve2GradientNN
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)
C++ interface for calculating backward pass gradient of 2D convolution This function calculates the g...
af_convolve2_gradient_nn
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)
C interface for calculating backward pass gradient of 2D convolution This function calculates the gra...
af_err
af_err
Definition: defines.h:71
defines.h