From 5ff67a26e9d50c488b38dd4de319875f044d9d3f Mon Sep 17 00:00:00 2001 From: Bhatu <prbhatu@microsoft.com> Date: Thu, 26 Nov 2020 13:21:10 +0530 Subject: [PATCH] Implement StopGradient as no-op --- Athos/TFCompiler/TFNodesAST.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/Athos/TFCompiler/TFNodesAST.py b/Athos/TFCompiler/TFNodesAST.py index 74cc02a..9b4fc5c 100644 --- a/Athos/TFCompiler/TFNodesAST.py +++ b/Athos/TFCompiler/TFNodesAST.py @@ -678,6 +678,10 @@ class TFNodesAST: return (None, AST.ID(dictNodeNameToOutVarStr[inputsRef[0]])) def Softmax(graph : Graph.Graph, curNode : Graph.Node, dictNodeNameToOutVarStr : dict, extraNodeInfoDict : dict): + inputsRef = curNode.getInputsRef() + return (None, AST.ID(dictNodeNameToOutVarStr[inputsRef[0]])) + + def StopGradient(graph : Graph.Graph, curNode : Graph.Node, dictNodeNameToOutVarStr : dict, extraNodeInfoDict : dict): inputsRef = curNode.getInputsRef() return (None, AST.ID(dictNodeNameToOutVarStr[inputsRef[0]])) -- GitLab