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

Commit 478b15e

Browse files
committed
updating to callAsFunction
1 parent 62b15ee commit 478b15e

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
@@ -536,7 +536,7 @@ public struct TransposedConv3D: Layer {
536536
/// - Parameter input: The input to the layer.
537537
/// - Returns: The output.
538538
@differentiable
539-
public func call(_ input: Tensor<Float>) -> Tensor<Float> {
539+
public func callAsFunction(_ input: Tensor<Float>) -> Tensor<Float> {
540540
let batchSize = input.shape[0]
541541
let w = (input.shape[1] - (1 * paddingIndex)) *
542542
strides.0 + (filter.shape[0] * paddingIndex)

0 commit comments

Comments
 (0)