TF fix ceildiv in _DimMixin.complete_dyn_size

rwth-i6 · Nov 7, 2023 · 9d34741 · 9d34741
1 parent 09f1048
commit 9d34741
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/returnn/tensor/_dim_extra.py b/returnn/tensor/_dim_extra.py
@@ -1184,6 +1184,8 @@ def _bin_op_tf(a, b):
             elif kind in ("floordiv", "truediv"):  # truediv assumes there is no remainder
                 return a // b
             elif kind == "ceildiv":
+                if util.is_onnx_export_global():
+                    return -tf_util.onnx_compat_floor_div(-a, b)
                 return -(-a // b)
             else:
                 raise ValueError("unknown op kind %r" % op.kind)