From 202da0ad7d9c0eb8718c4bf5b1e2ce3ca2d0536d 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: Wed, 21 Aug 2019 07:08:17 +0900 Subject: [PATCH] [moco-tf] Type inference for TFStopGradient (#6736) This will add TFStopGradient in type inference rule Signed-off-by: SaeHie Park --- compiler/moco-tf/src/Dialect/TFTypeInferenceRule.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/compiler/moco-tf/src/Dialect/TFTypeInferenceRule.cpp b/compiler/moco-tf/src/Dialect/TFTypeInferenceRule.cpp index ca07fa2..d905135 100644 --- a/compiler/moco-tf/src/Dialect/TFTypeInferenceRule.cpp +++ b/compiler/moco-tf/src/Dialect/TFTypeInferenceRule.cpp @@ -56,6 +56,7 @@ struct TypeForwardAlgorithm final : public moco::tf::TFNodeVisitorx()); } loco::DataType visit(const TFSquaredDifference *node) { return dtype_get(node->x()); } loco::DataType visit(const TFSqueeze *node) { return dtype_get(node->input()); } + loco::DataType visit(const TFStopGradient *node) { return dtype_get(node->input()); } loco::DataType visit(const TFSub *node) { return dtype_get(node->x()); } }; -- 2.7.4