diff --git a/mindspore/ops/_op_impl/tbe/batch_to_space.py b/mindspore/ops/_op_impl/tbe/batch_to_space.py
index 5d0bdc1de3b14bacf5704be0133d5aab8e44d3d0..c327419284a02fc33a1c863c406bd3f7731c251b 100644
--- a/mindspore/ops/_op_impl/tbe/batch_to_space.py
+++ b/mindspore/ops/_op_impl/tbe/batch_to_space.py
@@ -28,6 +28,7 @@ batch_to_space_op_info = TBERegOp("BatchToSpace") \
     .input(0, "x", False, "required", "all") \
     .output(0, "y", False, "required", "all") \
     .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
+    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
     .get_op_info()
 
 
diff --git a/mindspore/ops/_op_impl/tbe/space_to_batch.py b/mindspore/ops/_op_impl/tbe/space_to_batch.py
index d7c31edcbf2770a72b186c338f1b14652b8ea8ce..2a49077899b90ee379cea57f4793f26ce0d59f0a 100644
--- a/mindspore/ops/_op_impl/tbe/space_to_batch.py
+++ b/mindspore/ops/_op_impl/tbe/space_to_batch.py
@@ -28,6 +28,7 @@ space_to_batch_op_info = TBERegOp("SpaceToBatch") \
     .input(0, "x", False, "required", "all") \
     .output(0, "y", False, "required", "all") \
     .dtype_format(DataType.F16_5HD, DataType.F16_5HD) \
+    .dtype_format(DataType.F32_5HD, DataType.F32_5HD) \
     .get_op_info()