arrayfire/src/api/cpp/gradient.cpp at master · arrayfire/arrayfire · GitHub

Repository navigation

25 lines (21 loc) · 698 Bytes

/*******************************************************

* This file is distributed under 3-clause BSD license.

* The complete license agreement can be obtained at:

* http://arrayfire.com/licenses/BSD-3-Clause

********************************************************/

#include <af/array.h>

#include <af/image.h>

#include <utility>

#include "error.hpp"

namespace af {

void grad(array &rows, array &cols, const array &in) {

af_array rows_handle = 0;

af_array cols_handle = 0;

AF_THROW(af_gradient(&rows_handle, &cols_handle, in.get()));

rows = array(rows_handle);

cols = array(cols_handle);

}

} // namespace af