[moco/tf] name attribute for transforms (#4121)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Mon, 8 Jul 2019 01:06:31 +0000 (10:06 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Mon, 8 Jul 2019 01:06:31 +0000 (10:06 +0900)
This will add name() attribute for existing Transforms

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
contrib/moco-tf/src/Transforms/ClearAnnotTransform.h
contrib/moco-tf/src/Transforms/FixPaddingTransform.h
contrib/moco-tf/src/Transforms/FixShapeTransform.h

index 19bc6f8..ab56097 100644 (file)
@@ -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;
 };
 
index 439f333..4139c73 100644 (file)
@@ -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;
 };
 
index 39fc89f..d790d0e 100644 (file)
@@ -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;
 };