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

::typeof(op) -> op

This commit is contained in:
Dhairya Gandhi 2020-01-31 12:24:36 +05:30
parent a72ca2b05d
commit b9fbee1ff0
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/utils.jl
View File

@ -225,34 +225,34 @@ Base.copy(xs::Zeros{T,N}) where {T,N} = Zeros(T, size(xs)...)
# Define broadcasting behaviour # Define broadcasting behaviour
for op in (:+, :-) for op in (:+, :-)
@eval function broadcasted(::typeof($op), a::AbstractArray, b::Zeros) @eval function broadcasted($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(::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) broadcasted(-, a::Zeros, b::AbstractArray) = broadcasted(+, -b, a)
function broadcasted(::typeof(*), a::AbstractArray, b::Zeros) function broadcasted(*, 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(::typeof(*), a::Zeros, b::AbstractArray) = broadcasted(*, b, a) broadcasted(*, a::Zeros, b::AbstractArray) = broadcasted(*, b, a)
for op in (:+, :-, :*) for op in (:+, :-, :*)
@eval broadcasted(::typeof($op), a::Zeros, b::Zeros) = Zeros(Broadcast.broadcast_shape(size(a), size(b))...) @eval broadcasted($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(::typeof(+), a::AbstractArray, b::Zeros{T,0}) where T = a broadcasted(+, a::AbstractArray, b::Zeros{T,0}) where T = a
broadcasted(::typeof(+), a::Zeros{T,0}, b::AbstractArray) where T = b broadcasted(+, a::Zeros{T,0}, b::AbstractArray) where T = b
broadcasted(::typeof(-), a::AbstractArray, b::Zeros{T,0}) where T = a broadcasted(-, a::AbstractArray, b::Zeros{T,0}) where T = a
broadcasted(::typeof(-), a::Zeros{T,0}, b::AbstractArray) where T = -b broadcasted(-, a::Zeros{T,0}, b::AbstractArray) where T = -b
broadcasted(::typeof(*), a::AbstractArray, b::Zeros{T,0}) where T = zero(a) broadcasted(*, a::AbstractArray, b::Zeros{T,0}) where T = zero(a)
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) broadcasted(/, a::Zeros{T,0}, b::AbstractArray) where T = zero(b)
""" """
@jit ... @jit ...