diff --git a/DifferentiationInterfaceTest/ext/DifferentiationInterfaceTestFluxExt/DifferentiationInterfaceTestFluxExt.jl b/DifferentiationInterfaceTest/ext/DifferentiationInterfaceTestFluxExt/DifferentiationInterfaceTestFluxExt.jl index ee0d5b168..d0825cee3 100644 --- a/DifferentiationInterfaceTest/ext/DifferentiationInterfaceTestFluxExt/DifferentiationInterfaceTestFluxExt.jl +++ b/DifferentiationInterfaceTest/ext/DifferentiationInterfaceTestFluxExt/DifferentiationInterfaceTestFluxExt.jl @@ -10,11 +10,14 @@ using Flux: Conv, ConvTranspose, Dense, - GRU, GRUCell, - LSTM, LSTMCell, + GRU, + GRUCell, + LSTM, + LSTMCell, Maxout, MeanPool, - RNN, RNNCell, + RNN, + RNNCell, SamePad, Scale, SkipConnection,