TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Mike Innes 2018-03-05 23:44:25 +00:00
parent bfd6a4c0ec
commit 65ed95190a
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/tracker/Tracker.jl
View File

@ -1,5 +1,7 @@
module Tracker module Tracker
import Base: ==
export TrackedArray, TrackedVector, TrackedMatrix, param, back! export TrackedArray, TrackedVector, TrackedMatrix, param, back!
tracker(x) = nothing tracker(x) = nothing
@ -16,6 +18,9 @@ end
Call(f, args...) = Call{typeof(f),typeof(args)}(f, args) Call(f, args...) = Call{typeof(f),typeof(args)}(f, args)
# When deserialising, the object_id changes
a::Call == b::Call = a.func == b.func && a.args == b.args
@inline (c::Call)() = c.func(data.(c.args)...) @inline (c::Call)() = c.func(data.(c.args)...)
mutable struct Tracked{T} mutable struct Tracked{T}