7
7
8
8
function conv_nnpack! (y:: A1 , x:: A1 , w:: A1 , cdims:: ConvDims ;
9
9
b:: A2 = zeros (Float32, size (x, 3 )),
10
- algo = UInt32 (0 )):: A1 where {A1<: Array{Float32, 4} ,
10
+ algo = UInt32 (0 )) where {A1<: Array{Float32, 4} ,
11
11
A2<: Array{Float32, 1} }
12
12
check_dims (size (x), size (w), size (y), cdims)
13
13
threadpool = select_threadpool (cdims, size (y, 4 ))
@@ -21,7 +21,7 @@ function conv_nnpack!(y::A1, x::A1, w::A1, cdims::ConvDims;
21
21
end
22
22
23
23
function ∇conv_data_nnpack! (dx:: A , dy:: A , w:: A , cdims:: ConvDims ;
24
- algo = UInt32 (0 )):: A where {A<: Array{Float32, 4} }
24
+ algo = UInt32 (0 )) where {A<: Array{Float32, 4} }
25
25
check_dims (size (dx), size (w), size (dy), cdims)
26
26
threadpool = select_threadpool (cdims, size (y, 4 ))
27
27
@@ -34,7 +34,7 @@ function ∇conv_data_nnpack!(dx::A, dy::A, w::A, cdims::ConvDims;
34
34
end
35
35
36
36
function ∇conv_filter_nnpack! (dw:: A , x:: A , dy:: A , cdims:: ConvDims ;
37
- algo = UInt32 (0 )):: A where {A<: Array{Float32, 4} }
37
+ algo = UInt32 (0 )) where {A<: Array{Float32, 4} }
38
38
check_dims (size (x), size (dw), size (dy), cdims)
39
39
threadpool = select_threadpool (cdims, size (y, 4 ))
40
40
0 commit comments