MyCustomLayerPlugin Class Reference

Inheritance diagram for MyCustomLayerPlugin:

Collaboration diagram for MyCustomLayerPlugin:

Public Member Functions
bool	canHandle (const std::string &module_type) const override
std::unique_ptr< IModule >	createFromArchive (const std::string &device_type, const std::string &precision, ModelArchive &archive, const std::string &module_name, void *exec_context) const override
PluginInfo	getInfo () const override
void	registerOperations (OperationRegistry &registry) override

Member Function Documentation

canHandle()

bool MyCustomLayerPlugin::canHandle ( const std::string & module_type ) const

inlineoverride

createFromArchive()

std::unique_ptr< IModule > MyCustomLayerPlugin::createFromArchive ( const std::string & device_type, const std::string & precision, ModelArchive & archive, const std::string & module_name, void * exec_context ) const

inlineoverride

getInfo()

PluginInfo MyCustomLayerPlugin::getInfo ( ) const

inlineoverride

registerOperations()

void MyCustomLayerPlugin::registerOperations ( OperationRegistry & registry )

inlineoverride

---

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

• /__w/Mila/Mila/Mila/Src/Dnn/Extensibility/MyCustomPlugin.cpp