Merge 81f9fb3 into 09e7471

JuliaDiff · Mar 15, 2017 · 4241f08 · 4241f08
2 parents 09e7471 + 81f9fb3
commit 4241f08
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/api/utils.jl b/src/api/utils.jl
@@ -14,19 +14,18 @@ function seeded_reverse_pass!(result::AbstractArray, output::AbstractArray, inpu
 end
 
 function seeded_reverse_pass!(output::TrackedReal, input::TrackedReal, tape)
-    seed!(output)
     unseed!(input)
+    seed!(output)
     reverse_pass!(tape)
-    unseed!(output)
     return deriv(input)
 end
 
 # gradient (input is an array, output is a scalar) #
 #--------------------------------------------------#
 
 function seeded_reverse_pass!(result, output::TrackedReal, input, tape)
-    seed!(output)
     unseed!(input)
+    seed!(output)
     reverse_pass!(tape)
     extract_result!(result, output, input)
     return result
@@ -44,8 +43,8 @@ function seeded_reverse_pass!(result::AbstractArray, output::AbstractArray, inpu
     result_matrix = reshape(result, length(output), length(input))
     input_deriv = deriv(input)
     for i in eachindex(output)
-        seed!(output, i)
         unseed!(input)
+        seed!(output, i)
         reverse_pass!(tape)
         for j in eachindex(input)
             result_matrix[i, j] = input_deriv[j]