From eadae5b38fb2f5fb3217b6d5036e69af421d8185 Mon Sep 17 00:00:00 2001 From: Harsha Vardhan Simhadri Date: Wed, 31 Jul 2019 23:40:05 +0530 Subject: [PATCH] minor edits --- edgeml/pytorch/graph/rnn.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/edgeml/pytorch/graph/rnn.py b/edgeml/pytorch/graph/rnn.py index a67a9023..8d06437a 100644 --- a/edgeml/pytorch/graph/rnn.py +++ b/edgeml/pytorch/graph/rnn.py @@ -8,7 +8,8 @@ import numpy as np import edgeml.pytorch.utils as utils -def onnx_exportable_fastgrnn(input, fargs, output, hidden_size, wRank, uRank, gate_nonlinearity, update_nonlinearity): +def onnx_exportable_fastgrnn(input, fargs, output, hidden_size, wRank, uRank, + gate_nonlinearity, update_nonlinearity): class RNNSymbolic(Function): @staticmethod def symbolic(g, *fargs): @@ -25,8 +26,7 @@ def onnx_exportable_fastgrnn(input, fargs, output, hidden_size, wRank, uRank, ga def backward(ctx, *gargs, **gkwargs): raise RuntimeError("FIXME: Traced RNNs don't support backward") - output_temp = RNNSymbolic.apply(input, *fargs) - return output_temp + return RNNSymbolic.apply(input, *fargs) def gen_nonlinearity(A, nonlinearity): '''