Skip to content
Snippets Groups Projects
Commit 5ff67a26 authored by Bhatu's avatar Bhatu
Browse files

Implement StopGradient as no-op

parent 988fa41d
No related branches found
No related tags found
No related merge requests found
......@@ -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]]))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment