tflchef::TFliteOpRegistry Class Reference

tflchef operator registry More...

#include <TFliteOpRegistry.h>

Public Member Functions
const TFliteOpChef *	lookup (tflite::BuiltinOperator op) const
	Returns registered TFliteOpChef pointer for BuiltinOperator or nullptr if not registered.

Static Public Member Functions
static TFliteOpRegistry &	get ()

Detailed Description

tflchef operator registry

Definition at line 31 of file TFliteOpRegistry.h.

Member Function Documentation

get()

static TFliteOpRegistry & tflchef::TFliteOpRegistry::get ( )

inlinestatic

Definition at line 46 of file TFliteOpRegistry.h.

  {
    static TFliteOpRegistry me;
    return me;
  }

Referenced by tflchef::generate_recipe().

lookup()

const TFliteOpChef * tflchef::TFliteOpRegistry::lookup ( tflite::BuiltinOperator  op ) const

inline

Returns registered TFliteOpChef pointer for BuiltinOperator or nullptr if not registered.

Definition at line 38 of file TFliteOpRegistry.h.

  {
    if (_tfliteop_map.find(op) == _tfliteop_map.end())
      return nullptr;
 
    return _tfliteop_map.at(op).get();
  }

---

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

• compiler/tflchef/tflite/src/TFliteOpRegistry.h