luci::FuseMeanWithMeanPass Struct Reference

Class to fuse two Mean operations follow one by one into one Mean with merge reduction indices. More...

#include <FuseMeanWithMeanPass.h>

Collaboration diagram for luci::FuseMeanWithMeanPass:

Public Member Functions
const char *	name (void) const final
bool	run (loco::Graph *g) final
	Run the pass.
Public Member Functions inherited from logo::Pass
virtual	~Pass ()=default

Detailed Description

Class to fuse two Mean operations follow one by one into one Mean with merge reduction indices.

Definition at line 29 of file FuseMeanWithMeanPass.h.

Member Function Documentation

name()

const char * luci::FuseMeanWithMeanPass::name ( void  ) const

inlinefinalvirtual

Reimplemented from logo::Pass.

Definition at line 31 of file FuseMeanWithMeanPass.h.

{ return "luci::FuseMeanWithMeanPass"; }

run()

bool luci::FuseMeanWithMeanPass::run ( loco::Graph *  graph )

finalvirtual

Run the pass.

Returns

false if there was nothing changed

Implements logo::Pass.

Definition at line 156 of file FuseMeanWithMeanPass.cpp.

{
  bool changed = false;
  for (auto node : loco::active_nodes(loco::output_nodes(g)))
  {
    auto mean = dynamic_cast<luci::CircleMean *>(node);
    if (not mean)
      continue;
 
    if (fuse_mean_with_mean(mean))
      changed = true;
  }
 
  return changed;
}

References loco::active_nodes(), and loco::output_nodes().

Referenced by package.infer.session::inference().

---

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

• compiler/luci/pass/include/luci/Pass/FuseMeanWithMeanPass.h
• compiler/luci/pass/src/FuseMeanWithMeanPass.cpp