diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-02-23 06:48:28 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-23 06:48:28 +0100 |
| commit | eeb7e2b683bc514440bd464f78afc43d61dfe10b (patch) | |
| tree | 63032088ae331626ad4512d72b3235f2df0ad6b6 /candle-onnx | |
| parent | 11ea7aac4d6eb81c6e4d998f58252b7868a34e63 (diff) | |
| download | candle-eeb7e2b683bc514440bd464f78afc43d61dfe10b.tar.gz candle-eeb7e2b683bc514440bd464f78afc43d61dfe10b.tar.bz2 candle-eeb7e2b683bc514440bd464f78afc43d61dfe10b.zip | |
Apply rustfmt to the newly added tests. (#1749)
Diffstat (limited to 'candle-onnx')
| -rw-r--r-- | candle-onnx/tests/ops.rs | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/candle-onnx/tests/ops.rs b/candle-onnx/tests/ops.rs index 3d77071e..a686f198 100644 --- a/candle-onnx/tests/ops.rs +++ b/candle-onnx/tests/ops.rs @@ -861,11 +861,7 @@ fn test_shape_operation() -> Result<()> { sparse_initializer: vec![], quantization_annotation: vec![], })); - let x = Tensor::from_vec( - vec![1.0f32, 2.0f32, 3.0f32, 4.0f32], - &[2, 2], - &Device::Cpu, - )?; + let x = Tensor::from_vec(vec![1.0f32, 2.0f32, 3.0f32, 4.0f32], &[2, 2], &Device::Cpu)?; let mut inputs: HashMap<String, Tensor> = HashMap::new(); inputs.insert(INPUT_X.to_string(), x); @@ -1313,7 +1309,11 @@ fn test_relu_operation() -> Result<()> { sparse_initializer: 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(); inputs.insert(INPUT_X.to_string(), x); @@ -1334,4 +1334,4 @@ fn test_relu_operation() -> Result<()> { // #[test] // "Cast" -// #[test]
\ No newline at end of file +// #[test] |