BinaryArithmetic.h File Reference

#include "cker/Shape.h"
#include "cker/eigen/Utils.h"
#include "cker/operation/BroadcastTo.h"

Go to the source code of this file.

Namespaces
namespace	nnfw
namespace	nnfw::cker
namespace	nnfw::cker::train

Enumerations
enum class	nnfw::cker::train::ArithmeticType { nnfw::cker::train::kAdd , nnfw::cker::train::kSub , nnfw::cker::train::kMul , nnfw::cker::train::kDiv }

Functions
template<typename T >
void	nnfw::cker::train::BinaryArithmeticGrad (const Shape &lhs_shape, const T *lhs_data, const Shape &rhs_shape, const T *rhs_data, const Shape &incoming_shape, const T *incoming_data, const Shape &lhs_grad_shape, T *lhs_grad_data, const Shape &rhs_grad_shape, T *rhs_grad_data, ArithmeticType arithmetic_type)