onert::backend::trix::ops::BulkLayer Class Reference

#include <BulkLayer.h>

Collaboration diagram for onert::backend::trix::ops::BulkLayer:

Public Member Functions
	BulkLayer ()
	~BulkLayer ()
void	configure (const std::vector< const IPortableTensor * > &inputs, std::vector< IPortableTensor * > &outputs, std::string binary_path, const std::shared_ptr< DevContext > &dev_context)
void	run () override
void	prepare () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default

Detailed Description

Definition at line 34 of file BulkLayer.h.

Constructor & Destructor Documentation

BulkLayer()

onert::backend::trix::ops::BulkLayer::BulkLayer

(

)

Definition at line 30 of file BulkLayer.cc.

                     : _inputs(), _outputs(), _model_id(0), _dev_context(nullptr)
{
  // DO NOTHING
}

~BulkLayer()

onert::backend::trix::ops::BulkLayer::~BulkLayer

(

)

Definition at line 35 of file BulkLayer.cc.

{ _dev_context->unRegisterModel(_model_id); }

Member Function Documentation

configure()

void onert::backend::trix::ops::BulkLayer::configure	(	const std::vector< const IPortableTensor * > &	inputs,
		std::vector< IPortableTensor * > &	outputs,
		std::string	binary_path,
		const std::shared_ptr< DevContext > &	dev_context
	)

Definition at line 37 of file BulkLayer.cc.

{
  _inputs = inputs;
  _outputs = outputs;
  _dev_context = dev_context;
  _model_id = _dev_context->registerModel(binary_path);
}

prepare()

void onert::backend::trix::ops::BulkLayer::prepare ( )

overridevirtual

Reimplemented from onert::exec::IFunction.

Definition at line 68 of file BulkLayer.cc.

{
  // DO NOTHING
}

run()

void onert::backend::trix::ops::BulkLayer::run ( )

overridevirtual

Implements onert::exec::IFunction.

Definition at line 47 of file BulkLayer.cc.

{
  tensors_data_info in_info;
  tensors_data_info out_info;
  setDataInfo(_inputs, &in_info);
  setDataInfo(_outputs, &out_info);
 
  input_buffers input_bufs;
  output_buffers output_bufs;
  setBuffers(_inputs, &input_bufs);
  setBuffers(_outputs, &output_bufs);
 
  size_t batch_size = 1;
  // TODO Remove this assumption
  if (_inputs.size() == 1 && _outputs.size() == 1 && _inputs.at(0)->getShape().dim(0) > 1)
  {
    batch_size = _inputs.at(0)->getShape().dim(0);
  }
  _dev_context->requestRun(_model_id, &input_bufs, &in_info, &output_bufs, &out_info, batch_size);
}

References onert::backend::trix::setBuffers(), and onert::backend::trix::setDataInfo().

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

---

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

• runtime/onert/backend/trix/ops/BulkLayer.h
• runtime/onert/backend/trix/ops/BulkLayer.cc