1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
import _Differentiation
protocol Protocol: Differentiable {
// Test cross-file `@differentiable` attribute.
@differentiable(reverse, wrt: self)
func identityDifferentiableAttr() -> Self
// Test `@differentiable` propagation from storage declaration to accessors.
@differentiable(reverse)
var property: Float { get set }
// Test `@differentiable` propagation from storage declaration to accessors.
@differentiable(reverse)
subscript() -> Float { get set }
}
extension Protocol {
func identityDerivativeAttr() -> Self { self }
// Test cross-file `@derivative` attribute.
@derivative(of: identityDerivativeAttr)
func vjpIdentityDerivativeAttr() -> (
value: Self, pullback: (TangentVector) -> TangentVector
) {
fatalError()
}
}
class Class: Differentiable {
// Test `@differentiable` propagation from storage declaration to accessors.
@differentiable(reverse)
var property: Float {
get { 1 }
set {}
}
// Test `@differentiable` propagation from storage declaration to accessors.
@differentiable(reverse)
subscript() -> Float {
get { 1 }
set {}
}
}
struct S: Differentiable {
var value: Float
}
extension Array where Element == S {
@differentiable(reverse)
func sum() -> Float {
return 0
}
}
|
