add IntOrIntTuple

FluxML · Mar 25, 2021 · ce63fb3 · ce63fb3
1 parent 9fefa9f
commit ce63fb3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/NNlibCUDA/src/NNlibCUDA.jl b/lib/NNlibCUDA/src/NNlibCUDA.jl
@@ -4,6 +4,8 @@ using NNlib
 using CUDA
 using Random, Statistics
 
+const IntOrIntTuple = Union{Integer, NTuple{N,<:Integer} where N}
+
 include("upsample.jl")
 include("batchedmul.jl")
 include("scatter.jl")