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
|
// RUN: %target-build-swift %s
import _Differentiation
public protocol Layer: Differentiable {
associatedtype Input: Differentiable
associatedtype Output: Differentiable
@differentiable(reverse, wrt: (self, input))
@differentiable(reverse, wrt: input)
func callAsFunction(_ input: Input) -> Output
}
// Test for explicitly declared `@differentiable` attributes.
public class Function1<Input: Differentiable, Output: Differentiable>: Layer {
@noDerivative public let body: @differentiable(reverse) (Input) -> Output
public init(_ body: @escaping @differentiable(reverse) (Input) -> Output) {
self.body = body
}
@differentiable(reverse, wrt: (self, input))
@differentiable(reverse, wrt: input)
public func callAsFunction(_ input: Input) -> Output {
body(input)
}
@differentiable(reverse, wrt: x where T: Differentiable)
public func foo<T>(x: T) -> T {
x
}
}
// Test for implicitly generated `@differentiable` attributes.
class Function2<Input: Differentiable, Output: Differentiable>: Layer {
@noDerivative let body: @differentiable(reverse) (Input) -> Output
init(_ body: @escaping @differentiable(reverse) (Input) -> Output) {
self.body = body
}
func callAsFunction(_ input: Input) -> Output {
body(input)
}
}
|
