onert::backend::cpu::ops::RankLayer Class Reference

#include <RankLayer.h>

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

Public Member Functions
	RankLayer ()
void	configure (const IPortableTensor *input, IPortableTensor *output)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 33 of file RankLayer.h.

Constructor & Destructor Documentation

RankLayer()

onert::backend::cpu::ops::RankLayer::RankLayer

(

)

Definition at line 30 of file RankLayer.cc.

                     : _input(nullptr), _output(nullptr)
{
  // DO NOTHING
}

Member Function Documentation

configure()

void onert::backend::cpu::ops::RankLayer::configure	(	const IPortableTensor *	input,
		IPortableTensor *	output
	)

Definition at line 35 of file RankLayer.cc.

{
  _input = input;
  _output = output;
}

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 41 of file RankLayer.cc.

{
  int32_t *output_data = getBuffer<int32_t>(_output);
  output_data[0] = _input->getShape().rank();
}

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

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

---

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

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