* See the License for the specific language governing permissions and
* limitations under the License.
*/
-#include "TFLConverter.h"
+
+#include "CanonicalNodeConverter.h"
#include <loco.h>
#include <loco/IR/CanonicalDialect.h>
namespace exo
{
-template <typename CanonicalType> bool TFLConverter<CanonicalType>::run(loco::Graph *graph)
+template <typename CanonicalType>
+bool CanonicalNodeConverter<CanonicalType>::run(loco::Graph *graph)
{
auto active_nodes = loco::active_nodes(loco::output_nodes(graph));
bool changed = false;
// Add template instantiation of subclasses
// E.g.,
-// template bool TFLConverter<loco::AvgPool2D>::run(loco::Graph *graph);
+// template bool CanonicalNodeConverter<loco::AvgPool2D>::run(loco::Graph *graph);
} // namespace exo
* limitations under the License.
*/
-#ifndef __CONVERSION_TFLCONVERTER_H__
-#define __CONVERSION_TFLCONVERTER_H__
+#ifndef __CONVERSION_CANONICAL_NODE_CONVERTER_H__
+#define __CONVERSION_CANONICAL_NODE_CONVERTER_H__
#include "Convert.h"
/**
* @brief Class to convert a canonical node to TFL node
*/
-template <typename CanonicalType> class TFLConverter : public logo::Pass
+template <typename CanonicalType> class CanonicalNodeConverter : public logo::Pass
{
public:
virtual const char *name(void) const { return nullptr; }
} // namespace exo
-#endif //__CONVERSION_TFLCONVERTER_H__
+#endif //__CONVERSION_CANONICAL_NODE_CONVERTER_H__