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

Commit 133399b

Browse files
committed
updating to callAsFunction
1 parent bf2b462 commit 133399b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Sources/TensorFlow/Layers/Convolutional.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -535,7 +535,7 @@ public struct TransposedConv3D: Layer {
535535
/// - Parameter input: The input to the layer.
536536
/// - Returns: The output.
537537
@differentiable
538-
public func call(_ input: Tensor<Float>) -> Tensor<Float> {
538+
public func callAsFunction(_ input: Tensor<Float>) -> Tensor<Float> {
539539
let batchSize = input.shape[0]
540540
let w = (input.shape[1] - (1 * paddingIndex)) *
541541
strides.0 + (filter.shape[0] * paddingIndex)

0 commit comments

Comments
 (0)