|
75 | 75 | function ChainRulesCore.rrule(
|
76 | 76 | ::Type{
|
77 | 77 | <:ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10,
|
78 |
| - T11, T12, T13, T14 |
| 78 | + T11, T12, T13, T14, T15, T16 |
79 | 79 | }}, u,
|
80 | 80 | args...) where {T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11,
|
81 |
| - T12, T13, T14} |
| 81 | + T12, T13, T14, T15, T16} |
82 | 82 | function ODESolutionAdjoint(ȳ)
|
83 | 83 | (NoTangent(), ȳ, ntuple(_ -> NoTangent(), length(args))...)
|
84 | 84 | end
|
85 | 85 |
|
86 |
| - ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12, T13, T14}(u, args...), |
| 86 | + ODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, T11, T12, T13, T14, T15, T16}(u, args...), |
87 | 87 | ODESolutionAdjoint
|
88 | 88 | end
|
89 | 89 |
|
90 | 90 | function ChainRulesCore.rrule(
|
91 | 91 | ::Type{
|
92 |
| - <:RODESolution{uType, tType, isinplace, P, NP, F, G, K, |
93 |
| - ND |
| 92 | + <:RODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, |
| 93 | + T11, T12, T13, T14 |
94 | 94 | }}, u,
|
95 |
| - args...) where {uType, tType, isinplace, P, NP, F, G, K, ND} |
| 95 | + args...) where {T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, |
| 96 | + T11, T12, T13, T14} |
96 | 97 | function RODESolutionAdjoint(ȳ)
|
97 | 98 | (NoTangent(), ȳ, ntuple(_ -> NoTangent(), length(args))...)
|
98 | 99 | end
|
99 | 100 |
|
100 |
| - RODESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), |
| 101 | + RODESolution{T1, T2, T3, T4, T5, T6, T7, T8, T9, T10, |
| 102 | + T11, T12, T13, T14}(u, args...), |
101 | 103 | RODESolutionAdjoint
|
102 | 104 | end
|
103 | 105 |
|
|
0 commit comments