Skip to content

Commit

Permalink
Merge 78c353a into 56837ae
Browse files Browse the repository at this point in the history
  • Loading branch information
mohamed82008 committed Jul 18, 2020
2 parents 56837ae + 78c353a commit 34fa0c9
Showing 1 changed file with 0 additions and 29 deletions.
29 changes: 0 additions & 29 deletions src/derivatives/broadcast.jl
Original file line number Diff line number Diff line change
Expand Up @@ -101,35 +101,6 @@ function Base.copy(_bc::Broadcasted{TrackedStyle})
end
end

# https://github.com/FluxML/Flux.jl/issues/353
if VERSION < v"1.1.0-DEV.548"
@eval Base.Broadcast begin
function flatten(bc::Broadcasted{Style}) where {Style}
isflat(bc) && return bc
args = cat_nested(bc)
let makeargs = make_makeargs(bc), f = bc.f
newf = @inline function(args::Vararg{Any,N}) where N
f(makeargs(args...)...)
end
return Broadcasted{Style}(newf, args, bc.axes)
end
end
@inline function make_makeargs(makeargs, t::Tuple{<:Broadcasted,Vararg{Any}})
bc = t[1]
let makeargs = make_makeargs(makeargs, tail(t)), f = bc.f
let makeargs = make_makeargs(makeargs, bc.args)
headargs, tailargs = make_headargs(bc.args), make_tailargs(bc.args)
return @inline function(args::Vararg{Any,N}) where N
args1 = makeargs(args...)
a, b = headargs(args1...), tailargs(args1...)
(f(a...), b...)
end
end
end
end
end
end

getouttype(::TrackedReal{<:Any, D}) where {D} = D
getouttype(::TrackedArray{<:Any, D}) where {D} = D
getouttype(::Any) = Union{}
Expand Down

0 comments on commit 34fa0c9

Please sign in to comment.