also add compat for floordiv

rwth-i6 · Nov 7, 2023 · 69cb4f8 · 69cb4f8
1 parent b959efb
commit 69cb4f8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/returnn/tensor/_dim_extra.py b/returnn/tensor/_dim_extra.py
@@ -1182,6 +1182,8 @@ def _bin_op_tf(a, b):
             elif kind == "mul":
                 return a * b
             elif kind in ("floordiv", "truediv"):  # truediv assumes there is no remainder
+                if util.is_onnx_export_global():
+                    return tf_util.onnx_compat_floor_div(a, b)
                 return a // b
             elif kind == "ceildiv":
                 if util.is_onnx_export_global():