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

Commit 380a95c

Browse files
Shashi456rxwei
authored andcommitted
Update convolution operator parameters to have default values (#287)
1 parent b2e3699 commit 380a95c

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

Sources/TensorFlow/Operators/NN.swift

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -121,8 +121,8 @@ public extension Padding {
121121
public func conv2D<Scalar: TensorFlowFloatingPoint>(
122122
_ input: Tensor<Scalar>,
123123
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,
126126
dilations: (Int, Int, Int, Int) = (1, 1, 1, 1)
127127
) -> Tensor<Scalar> {
128128
return Raw.conv2D(
@@ -159,8 +159,8 @@ func conv2DBackpropInput<Scalar: TensorFlowFloatingPoint>(
159159
_ x: Tensor<Scalar>,
160160
shape: Tensor<Int32>,
161161
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,
164164
dilations: (Int, Int, Int, Int) = (1, 1, 1, 1)
165165
) -> Tensor<Scalar> {
166166
return Raw.conv2DBackpropInput(
@@ -198,8 +198,8 @@ func conv2DBackpropFilter<Scalar: TensorFlowFloatingPoint>(
198198
_ x: Tensor<Scalar>,
199199
input: Tensor<Scalar>,
200200
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,
203203
dilations: (Int, Int, Int, Int) = (1, 1, 1, 1)
204204
) -> Tensor<Scalar> {
205205
return Raw.conv2DBackpropFilter(
@@ -243,8 +243,8 @@ func _vjpConv2DBackpropFilter<Scalar: TensorFlowFloatingPoint>(
243243
public func conv3D<Scalar: TensorFlowFloatingPoint>(
244244
_ input: Tensor<Scalar>,
245245
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
248248
) -> Tensor<Scalar> {
249249
return Raw.conv3D(
250250
input,
@@ -280,8 +280,8 @@ func conv3DBackpropInput<Scalar: TensorFlowFloatingPoint>(
280280
_ x: Tensor<Scalar>,
281281
shape: Tensor<Int32>,
282282
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
285285
) -> Tensor<Scalar> {
286286
return Raw.conv3DBackpropInputV2(
287287
inputSizes: shape,
@@ -318,8 +318,8 @@ func conv3DBackpropFilter<Scalar: TensorFlowFloatingPoint>(
318318
_ x: Tensor<Scalar>,
319319
input: Tensor<Scalar>,
320320
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
323323
) -> Tensor<Scalar> {
324324
return Raw.conv3DBackpropFilterV2(
325325
x,

0 commit comments

Comments
 (0)