Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Co-authored-by: Guy Katz <guyk@cs.stanford.edu>
Co-authored-by: Kyle Julian <kjulian3.stanford.edu>
  • Loading branch information
kjulian3 and guykatzz authored Apr 13, 2020
1 parent 41ccc59 commit 98a0fde
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions maraboupy/MarabouNetworkONNX.py
Original file line number Diff line number Diff line change
Expand Up @@ -342,7 +342,7 @@ def transpose(self, node):
if inputName in self.varMap:
self.varMap[nodeName] = np.transpose(self.varMap[node.input[0]], perm)
elif inputName in self.constantMap:
self.constantMap[nodeName] = np.transpose(self.constant[inputName], perm)
self.constantMap[nodeName] = np.transpose(self.constantMap[inputName], perm)

def maxpoolEquations(self, node, makeEquations):
"""
Expand Down Expand Up @@ -812,4 +812,4 @@ def evaluateWithoutMarabou(self, inputValues):
printf("Not sure how to cast input to graph input of type %s" % onnxType)
raise NotImplementedError
input_dict[inputName] = inputValues[i].reshape(self.inputVars[i].shape).astype(inputType)
return sess.run([self.outputName],input_dict)[0]
return sess.run([self.outputName],input_dict)[0]

0 comments on commit 98a0fde

Please sign in to comment.