huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply rustfmt to the newly added tests. (#1749)

Browse Source
This commit is contained in:
Laurent Mazare
2024-02-23 06:48:28 +01:00
committed by GitHub
parent 11ea7aac4d
commit eeb7e2b683
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
candle-onnx/tests/ops.rs
View File

@ -861,11 +861,7 @@ fn test_shape_operation() -> Result<()> {
sparse_initializer: vec![], sparse_initializer: vec![],
quantization_annotation: vec![], quantization_annotation: vec![],
})); }));
let x = Tensor::from_vec( let x = Tensor::from_vec(vec![1.0f32, 2.0f32, 3.0f32, 4.0f32], &[2, 2], &Device::Cpu)?;
vec![1.0f32, 2.0f32, 3.0f32, 4.0f32],
&[2, 2],
&Device::Cpu,
)?;
let mut inputs: HashMap<String, Tensor> = HashMap::new(); let mut inputs: HashMap<String, Tensor> = HashMap::new();
inputs.insert(INPUT_X.to_string(), x); inputs.insert(INPUT_X.to_string(), x);
@ -1313,7 +1309,11 @@ fn test_relu_operation() -> Result<()> {
sparse_initializer: vec![], sparse_initializer: vec![],
quantization_annotation: vec![], quantization_annotation: vec![],
})); }));
let x = Tensor::from_vec(vec![-1.0f32, 1.0f32, -2.0f32, 3.0f32], &[2, 2], &Device::Cpu)?; let x = Tensor::from_vec(
vec![-1.0f32, 1.0f32, -2.0f32, 3.0f32],
&[2, 2],
&Device::Cpu,
)?;
let mut inputs: HashMap<String, Tensor> = HashMap::new(); let mut inputs: HashMap<String, Tensor> = HashMap::new();
inputs.insert(INPUT_X.to_string(), x); inputs.insert(INPUT_X.to_string(), x);