#include <ReshapeLayer.h>

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

Public Member Functions
	ReshapeLayer ()

void	reshapeGeneric ()

void	configure (const IPortableTensor input, const IPortableTensor shape, 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 ReshapeLayer.h.

Constructor & Destructor Documentation

◆ ReshapeLayer()

onert::backend::cpu::ops::ReshapeLayer::ReshapeLayer ( )

Definition at line 28 of file ReshapeLayer.cc.

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

Member Function Documentation

◆ configure()

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

Definition at line 43 of file ReshapeLayer.cc.

{
  _input = input;
  /* note : shape is optional. If not provided from model, _shape is nullptr. */
  _shape = shape;
  _output = output;
}

◆ reshapeGeneric()

void onert::backend::cpu::ops::ReshapeLayer::reshapeGeneric ( )

Definition at line 33 of file ReshapeLayer.cc.

{
  // output buffer equals to input buffer means that copy is not needed
  if (_output->buffer() != _input->buffer())
  {
    size_t count = _input->total_size();
    memcpy(_output->buffer(), _input->buffer(), count);
  }
}