diff options
Diffstat (limited to 'candle-nn/tests/batch_norm.rs')
| -rw-r--r-- | candle-nn/tests/batch_norm.rs | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/candle-nn/tests/batch_norm.rs b/candle-nn/tests/batch_norm.rs index 7a3cfc18..209fc10a 100644 --- a/candle-nn/tests/batch_norm.rs +++ b/candle-nn/tests/batch_norm.rs @@ -4,10 +4,8 @@ extern crate intel_mkl_src; #[cfg(feature = "accelerate")] extern crate accelerate_src; -mod test_utils; - use anyhow::Result; -use candle::{DType, Device, Tensor}; +use candle::{test_utils, DType, Device, Tensor}; use candle_nn::BatchNorm; /* The test below has been generated using the following PyTorch code: |