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.
1 parent b0f6676 commit 8ffb279Copy full SHA for 8ffb279
ext/RecursiveArrayToolsZygoteExt.jl
@@ -13,6 +13,8 @@ end
13
# Define a new species of projection operator for this type:
14
ChainRulesCore.ProjectTo(x::VectorOfArray) = ChainRulesCore.ProjectTo{VectorOfArray}()
15
16
+Zygote._tryreverse(m::typeof(map), x::AbstractVectorOfArray) = x
17
+
18
function ChainRulesCore.rrule(T::Type{<:RecursiveArrayTools.GPUArraysCore.AbstractGPUArray},
19
xs::AbstractVectorOfArray)
20
T(xs), ȳ -> (ChainRulesCore.NoTangent(), ȳ)
0 commit comments