@@ -121,8 +121,8 @@ public extension Padding {
121
121
public func conv2D< Scalar: TensorFlowFloatingPoint > (
122
122
_ input: Tensor < Scalar > ,
123
123
filter: Tensor < Scalar > ,
124
- strides: ( Int , Int , Int , Int ) ,
125
- padding: Padding ,
124
+ strides: ( Int , Int , Int , Int ) = ( 1 , 1 , 1 , 1 ) ,
125
+ padding: Padding = . valid ,
126
126
dilations: ( Int , Int , Int , Int ) = ( 1 , 1 , 1 , 1 )
127
127
) -> Tensor < Scalar > {
128
128
return Raw . conv2D (
@@ -159,8 +159,8 @@ func conv2DBackpropInput<Scalar: TensorFlowFloatingPoint>(
159
159
_ x: Tensor < Scalar > ,
160
160
shape: Tensor < Int32 > ,
161
161
filter: Tensor < Scalar > ,
162
- strides: ( Int , Int , Int , Int ) ,
163
- padding: Padding ,
162
+ strides: ( Int , Int , Int , Int ) = ( 1 , 1 , 1 , 1 ) ,
163
+ padding: Padding = . valid ,
164
164
dilations: ( Int , Int , Int , Int ) = ( 1 , 1 , 1 , 1 )
165
165
) -> Tensor < Scalar > {
166
166
return Raw . conv2DBackpropInput (
@@ -198,8 +198,8 @@ func conv2DBackpropFilter<Scalar: TensorFlowFloatingPoint>(
198
198
_ x: Tensor < Scalar > ,
199
199
input: Tensor < Scalar > ,
200
200
filterSizes: Tensor < Int32 > ,
201
- strides: ( Int , Int , Int , Int ) ,
202
- padding: Padding ,
201
+ strides: ( Int , Int , Int , Int ) = ( 1 , 1 , 1 , 1 ) ,
202
+ padding: Padding = . valid ,
203
203
dilations: ( Int , Int , Int , Int ) = ( 1 , 1 , 1 , 1 )
204
204
) -> Tensor < Scalar > {
205
205
return Raw . conv2DBackpropFilter (
@@ -243,8 +243,8 @@ func _vjpConv2DBackpropFilter<Scalar: TensorFlowFloatingPoint>(
243
243
public func conv3D< Scalar: TensorFlowFloatingPoint > (
244
244
_ input: Tensor < Scalar > ,
245
245
filter: Tensor < Scalar > ,
246
- strides: ( Int , Int , Int , Int , Int ) ,
247
- padding: Padding
246
+ strides: ( Int , Int , Int , Int , Int ) = ( 1 , 1 , 1 , 1 , 1 ) ,
247
+ padding: Padding = . valid
248
248
) -> Tensor < Scalar > {
249
249
return Raw . conv3D (
250
250
input,
@@ -280,8 +280,8 @@ func conv3DBackpropInput<Scalar: TensorFlowFloatingPoint>(
280
280
_ x: Tensor < Scalar > ,
281
281
shape: Tensor < Int32 > ,
282
282
filter: Tensor < Scalar > ,
283
- strides: ( Int , Int , Int , Int , Int ) ,
284
- padding: Padding
283
+ strides: ( Int , Int , Int , Int , Int ) = ( 1 , 1 , 1 , 1 , 1 ) ,
284
+ padding: Padding = . valid
285
285
) -> Tensor < Scalar > {
286
286
return Raw . conv3DBackpropInputV2 (
287
287
inputSizes: shape,
@@ -318,8 +318,8 @@ func conv3DBackpropFilter<Scalar: TensorFlowFloatingPoint>(
318
318
_ x: Tensor < Scalar > ,
319
319
input: Tensor < Scalar > ,
320
320
filterSizes: Tensor < Int32 > ,
321
- strides: ( Int , Int , Int , Int , Int ) ,
322
- padding: Padding
321
+ strides: ( Int , Int , Int , Int , Int ) = ( 1 , 1 , 1 , 1 , 1 ) ,
322
+ padding: Padding = . valid
323
323
) -> Tensor < Scalar > {
324
324
return Raw . conv3DBackpropFilterV2 (
325
325
x,
0 commit comments