We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
unthunk_tangent
1 parent a4d77ab commit ffc63c3Copy full SHA for ffc63c3
src/adjoint.jl
@@ -37,12 +37,6 @@ function unthunk_tangent end
37
@inline unthunk_tangent(x) = x
38
@inline unthunk_tangent(x::Tuple) = map(unthunk_tangent, x)
39
@inline unthunk_tangent(x::NamedTuple) = map(unthunk_tangent, x)
40
-@inline unthunk_tangent(x::AbstractThunk) = wrap_chainrules_output(unthunk(x))
41
-@inline unthunk_tangent(x::NTuple{N,<:Number}) where N = x
42
-@inline unthunk_tangent(x::AbstractArray{<:Number,N}) where N = x
43
-@inline unthunk_tangent(x::AbstractArray) = map(unthunk_tangent, x)
44
-unthunk_tangent(d::IdDict) = IdDict([unthunk_tangent(k) => unthunk_tangent(v) for (k, v) in d])
45
-@non_differentiable unthunk_tangent(::IdDict)
46
47
48
function gradm(ex, mut = false, keepthunks = false)
0 commit comments