luci::RemoveRedundantReshapePass Struct Reference

Class to remove redundant Reshape node into 1 Reshape node. More...

#include <RemoveRedundantReshapePass.h>

Collaboration diagram for luci::RemoveRedundantReshapePass:

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

Detailed Description

Class to remove redundant Reshape node into 1 Reshape node.

This class will update consecutive two Reshape node into single Reshape node. As Reshape operation just change shape, not buffer, former reshape could be unnecessary.

Definition at line 30 of file RemoveRedundantReshapePass.h.

Member Function Documentation

name()

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

inlinefinalvirtual

Reimplemented from logo::Pass.

Definition at line 32 of file RemoveRedundantReshapePass.h.

{ return "luci::RemoveRedundantReshapePass"; }

run()

bool luci::RemoveRedundantReshapePass::run ( loco::Graph *  g )

finalvirtual

BEFORE

 [CircleNode]
       |

[CircleReshape_1] | [CircleReshape_2] | [CircleNode]

AFTER

           [CircleNode]
           /          \

[CircleReshape_1] [CircleReshape_2] | [CircleNode]

Implements logo::Pass.

Definition at line 58 of file RemoveRedundantReshape.cpp.

{
  bool changed = false;
  for (auto node : loco::active_nodes(loco::output_nodes(g)))
  {
    if (auto reshape_node = dynamic_cast<luci::CircleReshape *>(node))
    {
      if (remove_redundant_reshape(reshape_node))
        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/RemoveRedundantReshapePass.h
• compiler/luci/pass/src/RemoveRedundantReshape.cpp