diff options
Diffstat (limited to 'candle-nn/tests/group_norm.rs')
| -rw-r--r-- | candle-nn/tests/group_norm.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/tests/group_norm.rs b/candle-nn/tests/group_norm.rs index f3ef2455..3a906c9d 100644 --- a/candle-nn/tests/group_norm.rs +++ b/candle-nn/tests/group_norm.rs @@ -23,7 +23,7 @@ extern crate intel_mkl_src; use anyhow::Result; use candle::{Device, Tensor}; -use candle_nn::GroupNorm; +use candle_nn::{GroupNorm, Module}; mod test_utils; use test_utils::to_vec3_round; |