Skip to content

Commit 6821859

Browse files
bgogulbgogul
committed
Fix style related issues from tensorflow#165 
1 parent 738b7a5 commit 6821859

File tree

4 files changed

+30
-15
lines changed

4 files changed

+30
-15
lines changed

‎Sources/TensorFlow/Core/TensorGroup.swift

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,8 @@ extension TensorHandle: TensorGroup {
101101
}
102102

103103
public init<C: RandomAccessCollection>(
104-
_handles: C) where C.Element == _AnyTensorHandle {
104+
_handles: C
105+
) where C.Element == _AnyTensorHandle {
105106
precondition(_handles.count == 1)
106107
self.init(handle: _handles[_handles.startIndex])
107108
}
@@ -129,7 +130,8 @@ extension ResourceHandle: TensorGroup {
129130
}
130131

131132
public init<C: RandomAccessCollection>(
132-
_handles: C) where C.Element == _AnyTensorHandle {
133+
_handles: C
134+
) where C.Element == _AnyTensorHandle {
133135
precondition(_handles.count == 1)
134136
self.init(handle: _handles[_handles.startIndex])
135137
}
@@ -157,7 +159,8 @@ extension VariantHandle: TensorGroup {
157159
}
158160

159161
public init<C: RandomAccessCollection>(
160-
_handles: C) where C.Element == _AnyTensorHandle {
162+
_handles: C
163+
) where C.Element == _AnyTensorHandle {
161164
precondition(_handles.count == 1)
162165
self.init(handle: _handles[_handles.startIndex])
163166
}
@@ -185,7 +188,8 @@ extension Tensor: TensorGroup {
185188
}
186189

187190
public init<C: RandomAccessCollection>(
188-
_handles: C) where C.Element == _AnyTensorHandle {
191+
_handles: C
192+
) where C.Element == _AnyTensorHandle {
189193
precondition(_handles.count == 1)
190194
self.init(handle: TensorHandle(handle: _handles[_handles.startIndex]))
191195
}
@@ -213,7 +217,8 @@ extension _TensorElementLiteral: TensorGroup {
213217
}
214218

215219
public init<C: RandomAccessCollection>(
216-
_handles: C) where C.Element == _AnyTensorHandle {
220+
_handles: C
221+
) where C.Element == _AnyTensorHandle {
217222
precondition(_handles.count == 1)
218223
self.init(handle: TensorHandle(handle: _handles[_handles.startIndex]))
219224
}
@@ -241,7 +246,8 @@ extension StringTensor: TensorGroup {
241246
}
242247

243248
public init<C: RandomAccessCollection>(
244-
_handles: C) where C.Element == _AnyTensorHandle {
249+
_handles: C
250+
) where C.Element == _AnyTensorHandle {
245251
precondition(_handles.count == 1)
246252
self.init(handle: TensorHandle(handle: _handles[_handles.startIndex]))
247253
}
@@ -283,7 +289,8 @@ extension Array: TensorArrayProtocol where Element: TensorGroup {
283289
}
284290

285291
public init<C: RandomAccessCollection>(
286-
_handles: C) where C.Element == _AnyTensorHandle {
292+
_handles: C
293+
) where C.Element == _AnyTensorHandle {
287294
let size = _handles.count / Int(Element._tensorHandleCount)
288295
self = (0..<size).map {
289296
let start = _handles.index(

‎Sources/TensorFlow/Operators/Dataset.swift

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,8 @@ public struct Zip2TensorGroup<T: TensorGroup, U: TensorGroup>: TensorGroup {
221221
}
222222

223223
public init<C: RandomAccessCollection>(
224-
_handles: C) where C.Element == _AnyTensorHandle {
224+
_handles: C
225+
) where C.Element == _AnyTensorHandle {
225226
let firstStart = _handles.startIndex
226227
let firstEnd = _handles.index(
227228
firstStart, offsetBy: Int(T._tensorHandleCount))

‎Tests/TensorFlowTests/OperatorTests/DatasetTests.swift

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@ struct SimpleOutput: TensorGroup {
2020
let b: TensorHandle<Int32>
2121

2222
public init<C: RandomAccessCollection>(
23-
_handles: C) where C.Element == _AnyTensorHandle {
23+
_handles: C
24+
) where C.Element == _AnyTensorHandle {
2425
precondition(_handles.count == 2)
2526
let aIndex = _handles.startIndex
2627
let bIndex = _handles.index(aIndex, offsetBy: 1)

‎Tests/TensorFlowTests/TensorGroupTests.swift

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,8 @@ extension TensorDataType : Equatable {
2525
struct Empty : TensorGroup {
2626
init() {}
2727
public init<C: RandomAccessCollection>(
28-
_handles: C) where C.Element == _AnyTensorHandle {}
28+
_handles: C
29+
) where C.Element == _AnyTensorHandle {}
2930
public var _tensorHandles: [_AnyTensorHandle] { [] }
3031
}
3132

@@ -38,7 +39,8 @@ struct Simple : TensorGroup, Equatable {
3839
}
3940

4041
public init<C: RandomAccessCollection>(
41-
_handles: C) where C.Element == _AnyTensorHandle {
42+
_handles: C
43+
) where C.Element == _AnyTensorHandle {
4244
precondition(_handles.count == 2)
4345
let wIndex = _handles.startIndex
4446
let bIndex = _handles.index(wIndex, offsetBy: 1)
@@ -61,7 +63,8 @@ struct Mixed : TensorGroup, Equatable {
6163
}
6264

6365
public init<C: RandomAccessCollection>(
64-
_handles: C) where C.Element == _AnyTensorHandle {
66+
_handles: C
67+
) where C.Element == _AnyTensorHandle {
6568
precondition(_handles.count == 2)
6669
let floatIndex = _handles.startIndex
6770
let intIndex = _handles.index(floatIndex, offsetBy: 1)
@@ -88,7 +91,8 @@ struct Nested : TensorGroup, Equatable {
8891
}
8992

9093
public init<C: RandomAccessCollection>(
91-
_handles: C) where C.Element == _AnyTensorHandle {
94+
_handles: C
95+
) where C.Element == _AnyTensorHandle {
9296
let simpleStart = _handles.startIndex
9397
let simpleEnd = _handles.index(
9498
simpleStart, offsetBy: Int(Simple._tensorHandleCount))
@@ -111,7 +115,8 @@ struct Generic<T: TensorGroup & Equatable, U: TensorGroup & Equatable> : TensorG
111115
}
112116

113117
public init<C: RandomAccessCollection>(
114-
_handles: C) where C.Element == _AnyTensorHandle {
118+
_handles: C
119+
) where C.Element == _AnyTensorHandle {
115120
let tStart = _handles.startIndex
116121
let tEnd = _handles.index(tStart, offsetBy: Int(T._tensorHandleCount))
117122
t = T.init(_handles: _handles[tStart..<tEnd])
@@ -134,7 +139,8 @@ struct UltraNested<T: TensorGroup & Equatable, V: TensorGroup & Equatable>
134139
}
135140

136141
public init<C: RandomAccessCollection>(
137-
_handles: C) where C.Element == _AnyTensorHandle {
142+
_handles: C
143+
) where C.Element == _AnyTensorHandle {
138144
let firstStart = _handles.startIndex
139145
let firstEnd = _handles.index(
140146
firstStart, offsetBy: Int(Generic<T,V>._tensorHandleCount))

0 commit comments

Comments
 (0)