huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 04:00:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move the test-utils bits to a shared place. (#619)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-27 09:42:22 +01:00
committed by GitHub
parent a8b39dd7b7
commit 5320aa6b7d
17 changed files with 34 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-nn/tests/batch_norm.rs
View File

@ -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: