loco::ShapeInferenceSession Class Reference

#include <ShapeInference.h>

Public Member Functions
	ShapeInferenceSession (const ShapeInferenceRule *rule)
bool	to (Graph *g) const

Detailed Description

Definition at line 36 of file ShapeInference.h.

Constructor & Destructor Documentation

ShapeInferenceSession()

loco::ShapeInferenceSession::ShapeInferenceSession ( const ShapeInferenceRule *  rule )

inline

Definition at line 39 of file ShapeInference.h.

                                                        : _rule{rule}
  {
    // DO NOTHING
  }

Member Function Documentation

to()

bool loco::ShapeInferenceSession::to

(

Graph *

g

)

const

Definition at line 68 of file ShapeInference.cpp.

{
  assert(_rule->support(ShapeInferenceRule::API::V1) && "API v1 is unavailable");
 
  bool changed = false;
 
  for (auto node : loco::postorder_traversal(loco::output_nodes(g)))
  {
    if (_rule->recognize(node->dialect()))
    {
      loco::NodeShape shape;
 
      if (!shape_known(node) && inputs_shape_ready(node))
      {
        if (_rule->infer(node, shape))
        {
          node->annot(std::make_unique<ShapeAnnotation>(shape));
          changed = true;
        }
      }
    }
  }
 
  return changed;
}

References loco::AnnotatedItem< Annotation >::annot(), loco::Node::dialect(), loco::ShapeInferenceRule::infer(), loco::output_nodes(), loco::postorder_traversal(), loco::ShapeInferenceRule::recognize(), loco::shape_known(), loco::ShapeInferenceRule::support(), and loco::ShapeInferenceRule::V1.

Referenced by exo::convert_to_TFLNodes(), exo::ShapeInferencePass::run(), exo::TypeInferencePass::run(), moco::tf::ShapeInferencePass::run(), and moco::tf::TypeInferencePass::run().

---

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

• compiler/loco/include/loco/Service/ShapeInference.h
• compiler/loco/src/Service/ShapeInference.cpp