diff --git a/docs/.gitignore b/docs/.gitignore index a303fff20..b71a83fb6 100644 --- a/docs/.gitignore +++ b/docs/.gitignore @@ -1,2 +1,3 @@ build/ site/ +Manifest.toml diff --git a/docs/Project.toml b/docs/Project.toml index 3a52a5db2..ce9aa3290 100644 --- a/docs/Project.toml +++ b/docs/Project.toml @@ -1,5 +1,3 @@ [deps] Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4" - -[compat] -Documenter = "0.27" +NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd" diff --git a/docs/make.jl b/docs/make.jl index a12937ca0..599aaf9dc 100644 --- a/docs/make.jl +++ b/docs/make.jl @@ -12,6 +12,7 @@ makedocs(modules = [NNlib], # analytics = "UA-36890222-9", assets = ["assets/flux.css"], prettyurls = get(ENV, "CI", nothing) == "true"), + warnonly=[:missing_docs,] ) deploydocs(repo = "github.com/FluxML/NNlib.jl.git", diff --git a/src/upsample.jl b/src/upsample.jl index babd613fb..5a60e8ec0 100644 --- a/src/upsample.jl +++ b/src/upsample.jl @@ -173,6 +173,16 @@ function upsample_nearest(x::AbstractArray{T,N}, scales::NTuple{S, <:Integer}) w out end +""" + ∇upsample_nearest(Δ::AbstractArray{T,3}, scales::NTuple{S, <:Integer}) where T + +# Arguments +- `Δ`: Incoming gradient array, backpropagated from downstream layers +- `scales`: scales by which the image was upsampled in the first place + +# Outputs +- `dx`: Downsampled version of `Δ` +""" function ∇upsample_nearest(x::AbstractArray{T,N}, scales::NTuple{S, <:Integer}) where {T,N,S} outsize = ntuple(N) do d d > S && return size(x,d)