From ce63fb3f21b1dfb1d0e1d9f41f29e6165ec3d0d0 Mon Sep 17 00:00:00 2001
From: Yueh-Hua Tu <a504082002@gmail.com>
Date: Wed, 17 Mar 2021 21:20:12 +0800
Subject: [PATCH] add IntOrIntTuple

---
 lib/NNlibCUDA/src/NNlibCUDA.jl | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lib/NNlibCUDA/src/NNlibCUDA.jl b/lib/NNlibCUDA/src/NNlibCUDA.jl
index 9f5009a0c..d6e05f2b9 100644
--- 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")