onert::backend::cpu::ops::CompareLayer Class Reference

#include <CompareLayer.h>

Collaboration diagram for onert::backend::cpu::ops::CompareLayer:

Public Member Functions
	CompareLayer ()
void	configure (const IPortableTensor *lhs, const IPortableTensor *rhs, const ir::operation::Comparison::ComparisonType op_type, IPortableTensor *output)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 34 of file CompareLayer.h.

Constructor & Destructor Documentation

CompareLayer()

onert::backend::cpu::ops::CompareLayer::CompareLayer

(

)

Definition at line 125 of file CompareLayer.cc.

  : _lhs(nullptr), _rhs(nullptr), _output(nullptr),
    _op_type(ir::operation::Comparison::ComparisonType::Equal)
{
  // DO NOTHING
}

Member Function Documentation

configure()

void onert::backend::cpu::ops::CompareLayer::configure	(	const IPortableTensor *	lhs,
		const IPortableTensor *	rhs,
		const ir::operation::Comparison::ComparisonType	op_type,
		IPortableTensor *	output
	)

Definition at line 132 of file CompareLayer.cc.

{
  _lhs = lhs;
  _rhs = rhs;
  _op_type = op_type;
  _output = output;
}

run()

void onert::backend::cpu::ops::CompareLayer::run ( )

overridevirtual

Implements onert::exec::IFunction.

Definition at line 141 of file CompareLayer.cc.

{
  if (_lhs->data_type() == OperandType::FLOAT32)
  {
    compareScalar<float>(_lhs, _rhs, _output, _op_type);
  }
  else if (_lhs->data_type() == OperandType::INT32)
  {
    compareScalar<int32_t>(_lhs, _rhs, _output, _op_type);
  }
  else if (_lhs->data_type() == OperandType::INT64)
  {
    compareScalar<int64_t>(_lhs, _rhs, _output, _op_type);
  }
  else if (_lhs->data_type() == OperandType::BOOL8)
  {
    compareScalar<uint8_t>(_lhs, _rhs, _output, _op_type);
  }
  else if (_lhs->data_type() == OperandType::QUANT_UINT8_ASYMM)
  {
    compareQuant8<uint8_t>(_lhs, _rhs, _output, _op_type);
  }
  else
  {
    throw std::runtime_error{"Compare: unsupported data type"};
  }
}

References onert::backend::IPortableTensor::data_type().

Referenced by package.infer.session::inference().

---

The documentation for this class was generated from the following files:

• runtime/onert/backend/cpu/ops/CompareLayer.h
• runtime/onert/backend/cpu/ops/CompareLayer.cc