Skip to content

Commit 24f9fc1

Browse files
committed
Make derivatives @usableFromInline + internal instead of public.
1 parent a615eb5 commit 24f9fc1

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

stdlib/public/Differentiation/DifferentiationSupport.swift

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1202,8 +1202,9 @@ extension Array : EuclideanDifferentiable
12021202
}
12031203

12041204
extension Array where Element : Differentiable {
1205+
@usableFromInline
12051206
@derivative(of: subscript)
1206-
public func _vjpSubscript(index: Int) ->
1207+
func _vjpSubscript(index: Int) ->
12071208
(value: Element, pullback: (Element.TangentVector) -> TangentVector)
12081209
{
12091210
func pullback(_ gradientIn: Element.TangentVector) -> TangentVector {
@@ -1216,8 +1217,9 @@ extension Array where Element : Differentiable {
12161217
return (self[index], pullback)
12171218
}
12181219

1220+
@usableFromInline
12191221
@derivative(of: +)
1220-
public static func _vjpPlus(_ lhs: [Element], _ rhs: [Element]) ->
1222+
static func _vjpConcatenate(_ lhs: [Element], _ rhs: [Element]) ->
12211223
(value: [Element], pullback: (TangentVector) -> (TangentVector, TangentVector)) {
12221224
func pullback(_ gradientIn: TangentVector) ->
12231225
(TangentVector, TangentVector) {
@@ -1237,17 +1239,19 @@ extension Array where Element : Differentiable {
12371239
}
12381240

12391241
extension Array where Element: Differentiable {
1242+
@usableFromInline
12401243
@derivative(of: append)
1241-
public mutating func _vjpAppend(_ element: Element) -> (
1244+
mutating func _vjpAppend(_ element: Element) -> (
12421245
value: Void, pullback: (inout TangentVector) -> Element.TangentVector
12431246
) {
12441247
let appendedElementIndex = count
12451248
defer { append(element) }
12461249
return ((), { dself in dself.base[appendedElementIndex] })
12471250
}
12481251

1252+
@usableFromInline
12491253
@derivative(of: append)
1250-
public mutating func _jvpAppend(_ element: Element) -> (
1254+
mutating func _jvpAppend(_ element: Element) -> (
12511255
value: Void, differential: (inout TangentVector, Element.TangentVector) -> Void
12521256
) {
12531257
append(element)

0 commit comments

Comments
 (0)