We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3db7900 commit b90c743Copy full SHA for b90c743
1 file changed
backends/cadence/aot/tests/test_ref_implementations.py
@@ -216,12 +216,12 @@ def test_quantized_linear(
216
expected_output: torch.Tensor,
217
) -> None:
218
src = (
219
- torch.arange(np.product(src_shape))
+ torch.arange(np.prod(src_shape))
220
.reshape(src_shape)
221
.to(expected_output.dtype)
222
)
223
weight = (
224
- torch.arange(np.product(weight_shape))
+ torch.arange(np.prod(weight_shape))
225
.reshape(weight_shape)
226
227
0 commit comments