GitHub Viewer
/*******************************************************
* Copyright (c) 2019, 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
********************************************************/
#include
#include
#include "error.hpp"
using af::array;
namespace af {
void meanvar(array& mean, array& var, const array& in, const array& weights,
const af_var_bias bias, const dim_t dim) {
af_array mean_ = mean.get();
af_array var_ = var.get();
AF_THROW(af_meanvar(&mean_, &var_, in.get(), weights.get(), bias, dim));
mean.set(mean_);
var.set(var_);
}
} // namespace af