diff options
| -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] |