From 5ea0ef6764c191a400ca27276e6435547ceca0b0 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Tue, 13 Feb 2018 16:15:36 +0000 Subject: [PATCH] tracker fix --- src/onehot.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/onehot.jl b/src/onehot.jl index b1a1a970..33c3921b 100644 --- a/src/onehot.jl +++ b/src/onehot.jl @@ -61,5 +61,5 @@ argmax(y::AbstractMatrix, l...) = # Ambiguity hack -a::TrackedMatrix * b::OneHotVector = TrackedArray(Tracker.Call(*, a, b)) -a::TrackedMatrix * b::OneHotMatrix = TrackedArray(Tracker.Call(*, a, b)) +a::TrackedMatrix * b::OneHotVector = Tracker.track(*, a, b) +a::TrackedMatrix * b::OneHotMatrix = Tracker.track(*, a, b)