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

missing conversions

Browse Source
This commit is contained in:
Mike J Innes 2019-02-14 13:30:27 +00:00
parent b0a5844afb
commit 54f7efa379
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/tracker/lib/array.jl
View File

@ -35,10 +35,10 @@ Base.eltype(x::Type{<:TrackedArray{T}}) where T <: Real = TrackedReal{T}
Base.convert(::Type{T}, x::S) where {T<:TrackedArray,S<:T} = x Base.convert(::Type{T}, x::S) where {T<:TrackedArray,S<:T} = x
Base.convert(::Type{<:TrackedArray}, x::TrackedArray) = Base.convert(::Type{TrackedArray{T,N,A}}, x::TrackedArray) where {T,N,A} =
error("Not implemented: convert $(typeof(x)) to $T") track(convert, A, x)
Base.convert(::Type{<:TrackedArray{T,N,A}}, x::AbstractArray) where {T,N,A} = Base.convert(::Type{TrackedArray{T,N,A}}, x::AbstractArray) where {T,N,A} =
TrackedArray(convert(A, x)) TrackedArray(convert(A, x))
Base.show(io::IO, t::Type{TrackedArray{T,N,A}}) where {T,N,A<:AbstractArray{T,N}} = Base.show(io::IO, t::Type{TrackedArray{T,N,A}}) where {T,N,A<:AbstractArray{T,N}} =

4
src/tracker/lib/real.jl
View File

@ -37,7 +37,9 @@ Base.convert(::Type{TrackedReal{T}}, x::TrackedReal{T}) where T = x
Base.convert(::Type{TrackedReal{T}}, x::Real) where T = TrackedReal(convert(T, x)) Base.convert(::Type{TrackedReal{T}}, x::Real) where T = TrackedReal(convert(T, x))
Base.convert(::Type{TrackedReal{T}}, x::TrackedReal{S}) where {T,S} = Base.convert(::Type{TrackedReal{T}}, x::TrackedReal{S}) where {T,S} =
error("Not implemented: convert tracked $S to tracked $T") track(convert, T, x)
@grad convert(T, x) = convert(T, data(x)), ȳ -> (nothing, convert(typeof(x), ȳ))
(T::Type{<:TrackedReal})(x::Real) = convert(T, x) (T::Type{<:TrackedReal})(x::Real) = convert(T, x)