Skip to content

Commit

Permalink
fix: bad rebase
Browse files Browse the repository at this point in the history
[skip tests]
  • Loading branch information
avik-pal committed Aug 22, 2024
1 parent 3f9fe2e commit f056a2d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/impl/batchnorm.jl
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,7 @@ function ∇batchnorm_affine_normalize(
end

function ∇batchnorm_affine_normalize!(
∂x::AbstractArray{∂xT, 3}, ∂σ²::AbstractArray{∂σ²T, 3}, ::Nothing,
∂x::AbstractArray{∂xT, 3}, ∂σ²::AbstractArray{∂σ²T, 3},
∂γ::Optional{<:AbstractArray{<:Any, 3}}, ::Union{GPUBroadcastOp, LoopedArrayOp},
∂y::AbstractArray{∂yT, 3}, x::AbstractArray{xT, 3}, μ::AbstractVector,
σ²::AbstractVector, γ::Optional{<:AbstractVector}, ϵ::Real,
Expand Down

0 comments on commit f056a2d

Please sign in to comment.