Skip to content
This repository was archived by the owner on Jul 1, 2023. It is now read-only.

Commit feb9d97

Browse files
authored
Fix ParameterlessLayer conformance. (#1035)
Follow-up to #1034.
1 parent 6ecfd32 commit feb9d97

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Tests/TensorFlowTests/LayerTests.swift

+2-2
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,10 @@ import XCTest
1717
@testable import TensorFlow
1818

1919
fileprivate struct Sigmoid<Scalar: TensorFlowFloatingPoint>: ParameterlessLayer {
20-
public init() {}
20+
typealias TangentVector = EmptyTangentVector
2121

2222
@differentiable
23-
public func callAsFunction(_ input: Tensor<Scalar>) -> Tensor<Scalar> {
23+
func callAsFunction(_ input: Tensor<Scalar>) -> Tensor<Scalar> {
2424
sigmoid(input)
2525
}
2626
}

0 commit comments

Comments
 (0)