locoex::TFLShapeInferenceRule Struct Reference

#include <TFLShapeInferenceRule.h>

Collaboration diagram for locoex::TFLShapeInferenceRule:

Public Member Functions
bool	recognize (const loco::Dialect *) const final
	Return true if this rule recognizes a given dialect.
bool	infer (const loco::Node *, loco::NodeShape &) const final
	Infer node's shape.
Public Member Functions inherited from loco::ShapeInferenceRule
virtual	~ShapeInferenceRule ()=default
virtual bool	support (const API &api) const
	Check whether a given API is available or not.
virtual void	infer (const Context *, const Node *, Sink *) const

Additional Inherited Members
Public Types inherited from loco::ShapeInferenceRule
enum class	API { V1 , V2 }

Detailed Description

Definition at line 25 of file TFLShapeInferenceRule.h.

Member Function Documentation

infer()

bool locoex::TFLShapeInferenceRule::infer ( const loco::Node *  , loco::NodeShape &    ) const

finalvirtual

Infer node's shape.

WARNING!!

Implementation SHOULD return true only when it succeeds in inference!

Implements loco::ShapeInferenceRule.

Definition at line 616 of file TFLShapeInferenceRule.cpp.

{
  assert(node->dialect() == TFLDialect::get());
  assert(dynamic_cast<const TFLNode *>(node) != nullptr);
 
  ShapeInferenceAlgorithm alg;
  shape = dynamic_cast<const TFLNode *>(node)->accept(&alg);
 
  return true;
}

References loco::Node::dialect().

recognize()

bool locoex::TFLShapeInferenceRule::recognize ( const loco::Dialect *  ) const

finalvirtual

Return true if this rule recognizes a given dialect.

Implements loco::ShapeInferenceRule.

Definition at line 611 of file TFLShapeInferenceRule.cpp.

{
  return TFLDialect::get() == d;
}

---

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

• compiler/exo/src/Dialect/Service/TFLShapeInferenceRule.h
• compiler/exo/src/Dialect/Service/TFLShapeInferenceRule.cpp