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

mat-vec mul

Browse Source
This commit is contained in:
Mike J Innes 2017-08-20 13:48:43 +01:00
parent 09974caba0
commit 65a49188e6
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/Tracker/lib.jl
View File

@ -14,7 +14,11 @@ a::TrackedMatrix * b::TrackedMatrix = TrackedArray(Call(*, a, b))
a::TrackedMatrix * b::AbstractMatrix = TrackedArray(Call(*, a, b))
a::AbstractMatrix * b::TrackedMatrix = TrackedArray(Call(*, a, b))
function back!(::typeof(*), Δ, a::AbstractMatrix, b::AbstractMatrix)
a::TrackedMatrix * b::TrackedVector = TrackedArray(Call(*, a, b))
a::TrackedMatrix * b::AbstractVector = TrackedArray(Call(*, a, b))
a::AbstractMatrix * b::TrackedVector = TrackedArray(Call(*, a, b))
function back!(::typeof(*), Δ, a::AbstractMatrix, b::AbstractVecOrMat)
@back!(a, A_mul_Bt(Δ, data(b)))
@back!(b, At_mul_B(data(a), Δ))
end