OperationUtils.cc File Reference

#include "OperationUtils.h"
#include <cker/eigen/redux_functor.h>
#include <cker/train/operation/ReLU.h>
#include <cker/train/operation/ReLU6.h>

Go to the source code of this file.

Namespaces
namespace	onert
namespace	onert::backend
namespace	onert::backend::train
namespace	onert::backend::train::ops

Functions
nnfw::cker::Shape	onert::backend::train::ops::getShape (const IPortableTensor *tensor)
	Get shape of tensor.
const IPortableTensor *	onert::backend::train::ops::backpropActivation (const ir::Activation &activation, const IPortableTensor *output, const IPortableTensor *input_backprop, IPortableTensor *output_backprop)
	backpropagate acitvation
void	onert::backend::train::ops::biasGrad (const IPortableTensor *input_backprop, IPortableTensor *bias_grad)
	backpropagate bias
nnfw::cker::train::LossReductionType	onert::backend::train::ops::convertLossReductionType (ir::train::LossReductionType type)
	convert loss reduction type