ONE - On-device Neural Engine
Loading...
Searching...
No Matches
exo::EltwiseMaxConverter Class Reference

Convert loco::EltwiseMax to TFLMaximum. More...

#include <EltwiseMaxConverter.h>

Collaboration diagram for exo::EltwiseMaxConverter:

Public Member Functions

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

Definition at line 30 of file EltwiseMaxConverter.h.

Member Function Documentation

◆ convert()

bool exo::EltwiseMaxConverter::convert ( loco::EltwiseMax origin)
finalvirtual

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

Definition at line 65 of file EltwiseMaxConverter.cpp.

66{
67 EltwiseMaxInputHandler input_handler;
69
70 auto tfl_new = domain_converter.convert<FeatureLayout::NHWC>(origin, input_handler);
71
72 return (tfl_new != nullptr);
73}
Class to handle domain conversion while converting a canonical node to TFL node(s)
Definition GraphBlock.h:135
TFLT * convert(CanonicalT *origin, InputHandler< CanonicalT, TFLT > &input_handler)
Performs domain conversion.
Definition GraphBlock.h:152

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

◆ name()

const char * exo::EltwiseMaxConverter::name ( void  ) const
inlinefinalvirtual

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

Definition at line 33 of file EltwiseMaxConverter.h.

33{ return "exo::EltwiseMaxConverter"; }

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