From 7f59c667e98ec9b96dc104f6ac08b684c7f118f5 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 8 Jul 2019 10:06:31 +0900 Subject: [PATCH] [moco/tf] name attribute for transforms (#4121) This will add name() attribute for existing Transforms Signed-off-by: SaeHie Park --- contrib/moco-tf/src/Transforms/ClearAnnotTransform.h | 3 +++ contrib/moco-tf/src/Transforms/FixPaddingTransform.h | 3 +++ contrib/moco-tf/src/Transforms/FixShapeTransform.h | 3 +++ 3 files changed, 9 insertions(+) diff --git a/contrib/moco-tf/src/Transforms/ClearAnnotTransform.h b/contrib/moco-tf/src/Transforms/ClearAnnotTransform.h index 19bc6f8..ab56097 100644 --- a/contrib/moco-tf/src/Transforms/ClearAnnotTransform.h +++ b/contrib/moco-tf/src/Transforms/ClearAnnotTransform.h @@ -32,6 +32,9 @@ namespace tf class ClearAnnotTransform : public Transform { public: + const char *name(void) const final { return "ClearAnnotTransform"; } + +public: bool run(loco::Graph *graph) override; }; diff --git a/contrib/moco-tf/src/Transforms/FixPaddingTransform.h b/contrib/moco-tf/src/Transforms/FixPaddingTransform.h index 439f333..4139c73 100644 --- a/contrib/moco-tf/src/Transforms/FixPaddingTransform.h +++ b/contrib/moco-tf/src/Transforms/FixPaddingTransform.h @@ -32,6 +32,9 @@ namespace tf class FixPaddingTransform : public Transform { public: + const char *name(void) const final { return "FixPaddingTransform"; } + +public: bool run(loco::Graph *graph) override; }; diff --git a/contrib/moco-tf/src/Transforms/FixShapeTransform.h b/contrib/moco-tf/src/Transforms/FixShapeTransform.h index 39fc89f..d790d0e 100644 --- a/contrib/moco-tf/src/Transforms/FixShapeTransform.h +++ b/contrib/moco-tf/src/Transforms/FixShapeTransform.h @@ -32,6 +32,9 @@ namespace tf class FixShapeTransform : public Transform { public: + const char *name(void) const final { return "FixShapeTransform"; } + +public: bool run(loco::Graph *graph) override; }; -- 2.7.4