diff --git a/Sources/TensorFlow/Core/TensorGroup.swift b/Sources/TensorFlow/Core/TensorGroup.swift index 8a84c4d0d..06acebb37 100644 --- a/Sources/TensorFlow/Core/TensorGroup.swift +++ b/Sources/TensorFlow/Core/TensorGroup.swift @@ -101,7 +101,8 @@ extension TensorHandle: TensorGroup { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { precondition(_handles.count == 1) self.init(handle: _handles[_handles.startIndex]) } @@ -129,7 +130,8 @@ extension ResourceHandle: TensorGroup { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { precondition(_handles.count == 1) self.init(handle: _handles[_handles.startIndex]) } @@ -157,7 +159,8 @@ extension VariantHandle: TensorGroup { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { precondition(_handles.count == 1) self.init(handle: _handles[_handles.startIndex]) } @@ -185,7 +188,8 @@ extension Tensor: TensorGroup { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { precondition(_handles.count == 1) self.init(handle: TensorHandle(handle: _handles[_handles.startIndex])) } @@ -213,7 +217,8 @@ extension _TensorElementLiteral: TensorGroup { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { precondition(_handles.count == 1) self.init(handle: TensorHandle(handle: _handles[_handles.startIndex])) } @@ -241,7 +246,8 @@ extension StringTensor: TensorGroup { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { precondition(_handles.count == 1) self.init(handle: TensorHandle(handle: _handles[_handles.startIndex])) } @@ -283,7 +289,8 @@ extension Array: TensorArrayProtocol where Element: TensorGroup { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { let size = _handles.count / Int(Element._tensorHandleCount) self = (0..: TensorGroup { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { let firstStart = _handles.startIndex let firstEnd = _handles.index( firstStart, offsetBy: Int(T._tensorHandleCount)) diff --git a/Tests/TensorFlowTests/OperatorTests/DatasetTests.swift b/Tests/TensorFlowTests/OperatorTests/DatasetTests.swift index edde5f886..9a3afb893 100644 --- a/Tests/TensorFlowTests/OperatorTests/DatasetTests.swift +++ b/Tests/TensorFlowTests/OperatorTests/DatasetTests.swift @@ -20,7 +20,8 @@ struct SimpleOutput: TensorGroup { let b: TensorHandle public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { precondition(_handles.count == 2) let aIndex = _handles.startIndex let bIndex = _handles.index(aIndex, offsetBy: 1) diff --git a/Tests/TensorFlowTests/TensorGroupTests.swift b/Tests/TensorFlowTests/TensorGroupTests.swift index 99d246098..58451cb53 100644 --- a/Tests/TensorFlowTests/TensorGroupTests.swift +++ b/Tests/TensorFlowTests/TensorGroupTests.swift @@ -25,7 +25,8 @@ extension TensorDataType : Equatable { struct Empty : TensorGroup { init() {} public init( - _handles: C) where C.Element == _AnyTensorHandle {} + _handles: C + ) where C.Element == _AnyTensorHandle {} public var _tensorHandles: [_AnyTensorHandle] { [] } } @@ -38,7 +39,8 @@ struct Simple : TensorGroup, Equatable { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { precondition(_handles.count == 2) let wIndex = _handles.startIndex let bIndex = _handles.index(wIndex, offsetBy: 1) @@ -61,7 +63,8 @@ struct Mixed : TensorGroup, Equatable { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { precondition(_handles.count == 2) let floatIndex = _handles.startIndex let intIndex = _handles.index(floatIndex, offsetBy: 1) @@ -88,7 +91,8 @@ struct Nested : TensorGroup, Equatable { } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { let simpleStart = _handles.startIndex let simpleEnd = _handles.index( simpleStart, offsetBy: Int(Simple._tensorHandleCount)) @@ -111,7 +115,8 @@ struct Generic : TensorG } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { let tStart = _handles.startIndex let tEnd = _handles.index(tStart, offsetBy: Int(T._tensorHandleCount)) t = T.init(_handles: _handles[tStart.. } public init( - _handles: C) where C.Element == _AnyTensorHandle { + _handles: C + ) where C.Element == _AnyTensorHandle { let firstStart = _handles.startIndex let firstEnd = _handles.index( firstStart, offsetBy: Int(Generic._tensorHandleCount))