diff options
Diffstat (limited to 'reference_model')
-rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index afd20e9..edc1793 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -1892,7 +1892,8 @@ int OpRFFT2d<Dtype>::eval() // Imaginary values with locations (0,0), (0,W/2), (H/2,0) and (H/2,W/2) are zero. // But due to sin(M_PI) not returning 0 because of M_PI being approximate, only // add to the imaginary sum when not processing these locations. - if ((ay % (half_in_height)) + (ax % (half_in_width)) > 0) + if ((in_height > 1 && (ay % (half_in_height)) > 0) || + (in_width > 1 && (ax % (half_in_width)) > 0)) { sum_imag += v_ir * a_sin; } |