17#ifndef __NNFW_CKER_TRAIN_OPERATION_REDUCEMEAN_H__
18#define __NNFW_CKER_TRAIN_OPERATION_REDUCEMEAN_H__
35 BroadcastTo(incoming_shape,
const_cast<T *
>(incoming_data), grad_shape, grad_data);
38 grad /= (grad.size() / incoming.size());
void MeanGrad(const Shape &incoming_shape, const T *incoming_data, const Shape &grad_shape, T *grad_data)
void BroadcastTo(const Shape &input_shape, T *input_data, const Shape &output_shape, T *output_data)
MatrixMap< Scalar > MapAsMatrixWithLastDimAsRows(Scalar *data, const Shape &shape)