exo::Relu6Converter Class Reference

Convert loco::Relu6 to TFLRelu6. More...

#include <Relu6Converter.h>

Collaboration diagram for exo::Relu6Converter:

Public Member Functions
const char *	name (void) const final
bool	convert (loco::ReLU6 *origin) final
Public Member Functions inherited from exo::CanonicalNodeConverter< loco::ReLU6 >
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::Relu6 to TFLRelu6.

Definition at line 30 of file Relu6Converter.h.

Member Function Documentation

convert()

bool exo::Relu6Converter::convert ( loco::ReLU6 *  origin )

finalvirtual

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

Definition at line 58 of file Relu6Converter.cpp.

{
  Relu6InputHandler input_handler;
  exo::DomainConverter<loco::ReLU6, locoex::TFLRelu6> 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::Relu6Converter::name ( void  ) const

inlinefinalvirtual

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

Definition at line 33 of file Relu6Converter.h.

{ return "exo::Relu6Converter"; }

---

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

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