exo::ReluConverter Class Reference

Convert loco::Relu to TFLRelu. More...

#include <ReluConverter.h>

Collaboration diagram for exo::ReluConverter:

Public Member Functions
const char *	name (void) const final
bool	convert (loco::ReLU *origin) final
Public Member Functions inherited from exo::CanonicalNodeConverter< loco::ReLU >
bool	run (loco::Graph *graph)
	Run the pass.
Public Member Functions inherited from logo::Pass
virtual	~Pass ()=default

Additional Inherited Members

Detailed Description

Convert loco::Relu to TFLRelu.

Definition at line 30 of file ReluConverter.h.

Member Function Documentation

convert()

bool exo::ReluConverter::convert ( loco::ReLU *  origin )

finalvirtual

Implements exo::CanonicalNodeConverter< loco::ReLU >.

Definition at line 58 of file ReluConverter.cpp.

{
  ReluInputHandler input_handler;
  exo::DomainConverter<loco::ReLU, locoex::TFLRelu> domain_converter;
 
  auto tfl_node = domain_converter.convert<FeatureLayout::NHWC>(origin, input_handler);
 
  return (tfl_node != nullptr);
}

References exo::DomainConverter< CanonicalT, TFLT >::convert(), and exo::NHWC.

name()

const char * exo::ReluConverter::name ( void  ) const

inlinefinalvirtual

Reimplemented from exo::CanonicalNodeConverter< loco::ReLU >.

Definition at line 33 of file ReluConverter.h.

{ return "exo::ReluConverter"; }

---

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

• compiler/exo/src/Conversion/ReluConverter.h
• compiler/exo/src/Conversion/ReluConverter.cpp