onert::backend::cpu::ops::FillLayer Class Reference

#include <FillLayer.h>

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

Public Member Functions
	FillLayer ()
void	configure (const IPortableTensor *value, 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 FillLayer.h.

Constructor & Destructor Documentation

FillLayer()

onert::backend::cpu::ops::FillLayer::FillLayer

(

)

Definition at line 32 of file FillLayer.cc.

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

Member Function Documentation

configure()

void onert::backend::cpu::ops::FillLayer::configure	(	const IPortableTensor *	value,
		IPortableTensor *	output
	)

Definition at line 37 of file FillLayer.cc.

{
  _value = value;
  _output = output;
}

run()

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

overridevirtual

Implements onert::exec::IFunction.

Definition at line 43 of file FillLayer.cc.

{
  switch (_output->data_type())
  {
    case OperandType::FLOAT32:
      nnfw::cker::Fill<float>(getBuffer<float>(_value), getShape(_output),
                              getBuffer<float>(_output));
      break;
    case OperandType::INT32:
      nnfw::cker::Fill<int32_t>(getBuffer<int32_t>(_value), getShape(_output),
                                getBuffer<int32_t>(_output));
      break;
    case OperandType::INT64:
      nnfw::cker::Fill<int64_t>(getBuffer<int64_t>(_value), getShape(_output),
                                getBuffer<int64_t>(_output));
      break;
    case OperandType::UINT32:
      nnfw::cker::Fill<uint32_t>(getBuffer<uint32_t>(_value), getShape(_output),
                                 getBuffer<uint32_t>(_output));
      break;
    default:
      throw std::runtime_error{"Fill: unsupported data type"};
  }
}

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

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

---

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

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