onert::backend::cpu::ops::ShapeLayer Class Reference

#include <ShapeLayer.h>

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

Public Member Functions
	ShapeLayer ()
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 ShapeLayer.h.

Constructor & Destructor Documentation

ShapeLayer()

onert::backend::cpu::ops::ShapeLayer::ShapeLayer

(

)

Definition at line 30 of file ShapeLayer.cc.

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

Member Function Documentation

configure()

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

Definition at line 44 of file ShapeLayer.cc.

{
  _input = input;
  _output = output;
}

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 50 of file ShapeLayer.cc.

{
  if (_output->data_type() == OperandType::UINT32)
  {
    GetRawShape(_input, getBuffer<uint32_t>(_output));
  }
  else if (_output->data_type() == OperandType::INT32)
  {
    GetRawShape(_input, getBuffer<int32_t>(_output));
  }
  else if (_output->data_type() == OperandType::INT64)
  {
    GetRawShape(_input, getBuffer<int64_t>(_output));
  }
  else
  {
    throw std::runtime_error{"NYI : not supported output type for ShapeLayer"};
  }
}

References onert::backend::IPortableTensor::data_type(), and onert::backend::cpu::ops::GetRawShape().

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

---

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

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