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

no-op copy

Browse Source
This commit is contained in:
Dhairya Gandhi 2020-01-31 13:23:33 +05:30
parent b9fbee1ff0
commit bc20103ea6
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/utils.jl
View File

@ -221,38 +221,38 @@ function *(a::Zeros{T,2}, b::AbstractArray{S,2}) where {T,S}
res .= zero(S) res .= zero(S)
end end
Base.copy(xs::Zeros{T,N}) where {T,N} = Zeros(T, size(xs)...) Base.copy(xs::Zeros{T,N}) where {T,N} = xs
# Define broadcasting behaviour # Define broadcasting behaviour
for op in (:+, :-) for op in (:+, :-)
@eval function broadcasted($op, a::AbstractArray, b::Zeros) @eval function broadcasted(::typeof($op), a::AbstractArray, b::Zeros)
sz = similar(a, Broadcast.broadcast_shape(size(a), size(b))) sz = similar(a, Broadcast.broadcast_shape(size(a), size(b)))
sz .= a sz .= a
end end
end end
broadcasted(+, a::Zeros, b::AbstractArray) = broadcasted(+, b, a) broadcasted(::typeof(+), a::Zeros, b::AbstractArray) = broadcasted(+, b, a)
broadcasted(-, a::Zeros, b::AbstractArray) = broadcasted(+, -b, a) broadcasted(::typeof(-), a::Zeros, b::AbstractArray) = broadcasted(+, -b, a)
function broadcasted(*, a::AbstractArray, b::Zeros) function broadcasted(::typeof(*), a::AbstractArray, b::Zeros)
sz = similar(a, Broadcast.broadcast_shape(size(a), size(b))) sz = similar(a, Broadcast.broadcast_shape(size(a), size(b)))
sz .= zero(a) sz .= zero(a)
end end
broadcasted(*, a::Zeros, b::AbstractArray) = broadcasted(*, b, a) broadcasted(::typeof(*), a::Zeros, b::AbstractArray) = broadcasted(*, b, a)
for op in (:+, :-, :*) for op in (:+, :-, :*)
@eval broadcasted($op, a::Zeros, b::Zeros) = Zeros(Broadcast.broadcast_shape(size(a), size(b))...) @eval broadcasted(::typeof($op), a::Zeros, b::Zeros) = Zeros(Broadcast.broadcast_shape(size(a), size(b))...)
end end
# Some opportunities to avoid scalar indexing, intermediaries # Some opportunities to avoid scalar indexing, intermediaries
broadcasted(+, a::AbstractArray, b::Zeros{T,0}) where T = a broadcasted(::typeof(+), a::AbstractArray, b::Zeros{T,0}) where T = a
broadcasted(+, a::Zeros{T,0}, b::AbstractArray) where T = b broadcasted(::typeof(+), a::Zeros{T,0}, b::AbstractArray) where T = b
broadcasted(-, a::AbstractArray, b::Zeros{T,0}) where T = a broadcasted(::typeof(-), a::AbstractArray, b::Zeros{T,0}) where T = a
broadcasted(-, a::Zeros{T,0}, b::AbstractArray) where T = -b broadcasted(::typeof(-), a::Zeros{T,0}, b::AbstractArray) where T = -b
broadcasted(*, a::AbstractArray, b::Zeros{T,0}) where T = zero(a) broadcasted(::typeof(*), a::AbstractArray, b::Zeros{T,0}) where T = zero(a)
broadcasted(*, a::Zeros{T,0}, b::AbstractArray) where T = zero(b) broadcasted(::typeof(*), a::Zeros{T,0}, b::AbstractArray) where T = zero(b)
broadcasted(/, a::Zeros{T,0}, b::AbstractArray) where T = zero(b) broadcasted(::typeof(/), a::Zeros{T,0}, b::AbstractArray) where T = zero(b)
""" """
@jit ... @jit ...