|
16 | 16 | @frozen
|
17 | 17 | public struct MaxPool1D<Scalar: TensorFlowFloatingPoint>: ParameterlessLayer {
|
18 | 18 | /// The size of the sliding reduction window for pooling.
|
19 |
| - @noDerivative let poolSize: Int |
| 19 | + @noDerivative public let poolSize: Int |
20 | 20 | /// The stride of the sliding window for temporal dimension.
|
21 |
| - @noDerivative let stride: Int |
| 21 | + @noDerivative public let stride: Int |
22 | 22 | /// The padding algorithm for pooling.
|
23 |
| - @noDerivative let padding: Padding |
| 23 | + @noDerivative public let padding: Padding |
24 | 24 |
|
25 | 25 | /// Creates a max pooling layer.
|
26 | 26 | ///
|
@@ -53,12 +53,12 @@ public struct MaxPool1D<Scalar: TensorFlowFloatingPoint>: ParameterlessLayer {
|
53 | 53 | @frozen
|
54 | 54 | public struct MaxPool2D<Scalar: TensorFlowFloatingPoint>: ParameterlessLayer {
|
55 | 55 | /// The size of the sliding reduction window for pooling.
|
56 |
| - @noDerivative let poolSize: (Int, Int, Int, Int) |
| 56 | + @noDerivative public let poolSize: (Int, Int, Int, Int) |
57 | 57 | /// The strides of the sliding window for each dimension of a 4-D input.
|
58 | 58 | /// Strides in non-spatial dimensions must be `1`.
|
59 |
| - @noDerivative let strides: (Int, Int, Int, Int) |
| 59 | + @noDerivative public let strides: (Int, Int, Int, Int) |
60 | 60 | /// The padding algorithm for pooling.
|
61 |
| - @noDerivative let padding: Padding |
| 61 | + @noDerivative public let padding: Padding |
62 | 62 |
|
63 | 63 | /// Creates a max pooling layer.
|
64 | 64 | public init(poolSize: (Int, Int, Int, Int), strides: (Int, Int, Int, Int), padding: Padding) {
|
@@ -96,12 +96,12 @@ public extension MaxPool2D {
|
96 | 96 | @frozen
|
97 | 97 | public struct MaxPool3D<Scalar: TensorFlowFloatingPoint>: ParameterlessLayer {
|
98 | 98 | /// The size of the sliding reduction window for pooling.
|
99 |
| - @noDerivative let poolSize: (Int, Int, Int, Int, Int) |
| 99 | + @noDerivative public let poolSize: (Int, Int, Int, Int, Int) |
100 | 100 | /// The strides of the sliding window for each dimension of a 5-D input.
|
101 | 101 | /// Strides in non-spatial dimensions must be `1`.
|
102 |
| - @noDerivative let strides: (Int, Int, Int, Int, Int) |
| 102 | + @noDerivative public let strides: (Int, Int, Int, Int, Int) |
103 | 103 | /// The padding algorithm for pooling.
|
104 |
| - @noDerivative let padding: Padding |
| 104 | + @noDerivative public let padding: Padding |
105 | 105 |
|
106 | 106 | /// Creates a max pooling layer.
|
107 | 107 | public init(
|
@@ -154,11 +154,11 @@ public extension MaxPool3D {
|
154 | 154 | @frozen
|
155 | 155 | public struct AvgPool1D<Scalar: TensorFlowFloatingPoint>: ParameterlessLayer {
|
156 | 156 | /// The size of the sliding reduction window for pooling.
|
157 |
| - @noDerivative let poolSize: Int |
| 157 | + @noDerivative public let poolSize: Int |
158 | 158 | /// The stride of the sliding window for temporal dimension.
|
159 |
| - @noDerivative let stride: Int |
| 159 | + @noDerivative public let stride: Int |
160 | 160 | /// The padding algorithm for pooling.
|
161 |
| - @noDerivative let padding: Padding |
| 161 | + @noDerivative public let padding: Padding |
162 | 162 |
|
163 | 163 | /// Creates an average pooling layer.
|
164 | 164 | ///
|
@@ -191,12 +191,12 @@ public struct AvgPool1D<Scalar: TensorFlowFloatingPoint>: ParameterlessLayer {
|
191 | 191 | @frozen
|
192 | 192 | public struct AvgPool2D<Scalar: TensorFlowFloatingPoint>: ParameterlessLayer {
|
193 | 193 | /// The size of the sliding reduction window for pooling.
|
194 |
| - @noDerivative let poolSize: (Int, Int, Int, Int) |
| 194 | + @noDerivative public let poolSize: (Int, Int, Int, Int) |
195 | 195 | /// The strides of the sliding window for each dimension of a 4-D input.
|
196 | 196 | /// Strides in non-spatial dimensions must be `1`.
|
197 |
| - @noDerivative let strides: (Int, Int, Int, Int) |
| 197 | + @noDerivative public let strides: (Int, Int, Int, Int) |
198 | 198 | /// The padding algorithm for pooling.
|
199 |
| - @noDerivative let padding: Padding |
| 199 | + @noDerivative public let padding: Padding |
200 | 200 |
|
201 | 201 | /// Creates an average pooling layer.
|
202 | 202 | public init(poolSize: (Int, Int, Int, Int), strides: (Int, Int, Int, Int), padding: Padding) {
|
@@ -234,12 +234,12 @@ public extension AvgPool2D {
|
234 | 234 | @frozen
|
235 | 235 | public struct AvgPool3D<Scalar: TensorFlowFloatingPoint>: ParameterlessLayer {
|
236 | 236 | /// The size of the sliding reduction window for pooling.
|
237 |
| - @noDerivative let poolSize: (Int, Int, Int, Int, Int) |
| 237 | + @noDerivative public let poolSize: (Int, Int, Int, Int, Int) |
238 | 238 | /// The strides of the sliding window for each dimension of a 5-D input.
|
239 | 239 | /// Strides in non-spatial dimensions must be `1`.
|
240 |
| - @noDerivative let strides: (Int, Int, Int, Int, Int) |
| 240 | + @noDerivative public let strides: (Int, Int, Int, Int, Int) |
241 | 241 | /// The padding algorithm for pooling.
|
242 |
| - @noDerivative let padding: Padding |
| 242 | + @noDerivative public let padding: Padding |
243 | 243 |
|
244 | 244 | /// Creates an average pooling layer.
|
245 | 245 | public init(
|
|
0 commit comments