ONE - On-device Neural Engine
Loading...
Searching...
No Matches
BinaryArithmeticOps.h File Reference
#include "cker/Shape.h"
#include "cker/Types.h"
#include "cker/Utils.h"
#include <cmath>

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 
namespace  nnfw::cker::reference
 

Functions

template<typename T >
void nnfw::cker::reference::BinaryArithmeticOp (const BinaryArithmeticOpParam &params, const Shape &input1_shape, const T *input1_data, const Shape &input2_shape, const T *input2_data, const Shape &output_shape, T *output_data, const std::function< T(const T &, const T &)> &fn)
 
template<>
void nnfw::cker::reference::BinaryArithmeticOp (const BinaryArithmeticOpParam &params, const Shape &input1_shape, const float *input1_data, const Shape &input2_shape, const float *input2_data, const Shape &output_shape, float *output_data, const std::function< float(const float &, const float &)> &fn)
 
template<>
void nnfw::cker::reference::BinaryArithmeticOp (const BinaryArithmeticOpParam &, const Shape &input1_shape, const bool *input1_data, const Shape &input2_shape, const bool *input2_data, const Shape &output_shape, bool *output_data, const std::function< bool(const bool &, const bool &)> &fn)
 
template<>
void nnfw::cker::reference::BinaryArithmeticOp (const BinaryArithmeticOpParam &params, const Shape &input1_shape, const int64_t *input1_data, const Shape &input2_shape, const int64_t *input2_data, const Shape &output_shape, int64_t *output_data, const std::function< int64_t(const int64_t &, const int64_t &)> &fn)
 
template<typename T >
std::enable_if_t< is_quant8< T >::value > nnfw::cker::reference::BroadcastBinaryArithmeticOpSlow (const BinaryArithmeticOpParam &params, const Shape &input1_shape, const T *input1_data, const Shape &input2_shape, const T *input2_data, const Shape &output_shape, T *output_data, const std::function< T(const BinaryArithmeticOpParam &params, const T &, const T &)> &fn)
 
template<typename T >
void nnfw::cker::reference::BroadcastBinaryArithmeticOpSlow (const BinaryArithmeticOpParam &params, const Shape &input1_shape, const T *input1_data, const Shape &input2_shape, const T *input2_data, const Shape &output_shape, T *output_data, const std::function< T(const T &, const T &)> &fn)
 
template<>
void nnfw::cker::reference::BroadcastBinaryArithmeticOpSlow (const BinaryArithmeticOpParam &params, const Shape &input1_shape, const float *input1_data, const Shape &input2_shape, const float *input2_data, const Shape &output_shape, float *output_data, const std::function< float(const float &, const float &)> &fn)
 
template<>
void nnfw::cker::reference::BroadcastBinaryArithmeticOpSlow (const BinaryArithmeticOpParam &, const Shape &input1_shape, const bool *input1_data, const Shape &input2_shape, const bool *input2_data, const Shape &output_shape, bool *output_data, const std::function< bool(const bool &, const bool &)> &fn)
 
template<>
void nnfw::cker::reference::BroadcastBinaryArithmeticOpSlow (const BinaryArithmeticOpParam &params, const Shape &input1_shape, const int64_t *input1_data, const Shape &input2_shape, const int64_t *input2_data, const Shape &output_shape, int64_t *output_data, const std::function< int64_t(const int64_t &, const int64_t &)> &fn)