exo::TensorReduceConverter Class Reference

Convert loco::TensorReduce to appropriate TFL reduce operation. More...

#include <TensorReduceConverter.h>

Collaboration diagram for exo::TensorReduceConverter:

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

Note

loco::TensorReduce always keep dimensions

Currently support:

• When loco::TensorReduce::func() == Mean, convert to TFLMean + TFLConst
• TODO Support other cases

Definition at line 35 of file TensorReduceConverter.h.

Member Function Documentation

convert()

bool exo::TensorReduceConverter::convert ( loco::TensorReduce *  origin )

finalvirtual

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

Definition at line 87 of file TensorReduceConverter.cpp.

{
  if (origin->func() == loco::ReduceFunc::Mean)
    return convert_as_mean(origin);
  else
    INTERNAL_EXN_V("Unsupported ReduceFunc", oops::to_uint32(origin->func()));
}

References loco::TensorReduce::func(), INTERNAL_EXN_V, loco::Mean, and oops::to_uint32().

name()

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

inlinefinalvirtual

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

Definition at line 38 of file TensorReduceConverter.h.

{ return "exo::TensorReduceConverter"; }

---

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

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