AthOnnx::OnnxRuntimeSvc Class Reference

Service implementing AthOnnx::IOnnxRuntimeSvc. More...

#include <OnnxRuntimeSvc.h>

Inheritance diagram for AthOnnx::OnnxRuntimeSvc:

Collaboration diagram for AthOnnx::OnnxRuntimeSvc:

Public Member Functions
Function(s) inherited from @c Service
	OnnxRuntimeSvc (const std::string &name, ISvcLocator *svc)
virtual StatusCode	initialize () override
	Function initialising the service. More...
virtual StatusCode	finalize () override
	Function finalising the service. More...
Function(s) inherited from @c AthOnnx::IOnnxRuntimeSvc
virtual Ort::Env &	env () const override
	Return the Onnx Runtime environment object. More...

Private Attributes
std::unique_ptr< Ort::Env >	m_env
	Global runtime environment for Onnx Runtime. More...

Detailed Description

Service implementing AthOnnx::IOnnxRuntimeSvc.

This is a very simple implementation, just managing the lifetime of some Onnx Runtime C++ objects.

Author

Attila Krasznahorkay Attil.nosp@m.a.Kr.nosp@m.aszna.nosp@m.hork.nosp@m.ay@ce.nosp@m.rn.c.nosp@m.h

Definition at line 27 of file OnnxRuntimeSvc.h.

Constructor & Destructor Documentation

OnnxRuntimeSvc()

AthOnnx::OnnxRuntimeSvc::OnnxRuntimeSvc	(	const std::string &	name,
		ISvcLocator *	svc
	)

Definition at line 8 of file OnnxRuntimeSvc.cxx.

                                                                        :
      base_class(name, svc)
   {
   }

Member Function Documentation

env()

Ort::Env & AthOnnx::OnnxRuntimeSvc::env ( ) const

overridevirtual

Return the Onnx Runtime environment object.

Definition at line 38 of file OnnxRuntimeSvc.cxx.

                                     {
 
      return *m_env;
   }

finalize()

StatusCode AthOnnx::OnnxRuntimeSvc::finalize ( )

overridevirtual

Function finalising the service.

Definition at line 28 of file OnnxRuntimeSvc.cxx.

                                       {
 
      // Dekete the environment object.
      m_env.reset();
      ATH_MSG_DEBUG( "Ort::Env object deleted" );
 
      // Return gracefully.
      return StatusCode::SUCCESS;
   }

initialize()

StatusCode AthOnnx::OnnxRuntimeSvc::initialize ( )

overridevirtual

Function initialising the service.

Definition at line 13 of file OnnxRuntimeSvc.cxx.

                                         {
 
      // Create the environment object.
      Ort::ThreadingOptions tp_options;
      tp_options.SetGlobalIntraOpNumThreads(1);
      tp_options.SetGlobalInterOpNumThreads(1);
 
      m_env = std::make_unique< Ort::Env >(
            tp_options, ORT_LOGGING_LEVEL_WARNING, name().c_str());
      ATH_MSG_DEBUG( "Ort::Env object created" );
 
      // Return gracefully.
      return StatusCode::SUCCESS;
   }

Member Data Documentation

m_env

std::unique_ptr< Ort::Env > AthOnnx::OnnxRuntimeSvc::m_env

private

Global runtime environment for Onnx Runtime.

Definition at line 52 of file OnnxRuntimeSvc.h.

---

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

• OnnxRuntimeSvc.h
• OnnxRuntimeSvc.cxx