diff --git a/mindspore/ops/_op_impl/tbe/batch_to_space.py b/mindspore/ops/_op_impl/tbe/batch_to_space.py
index 5d0bdc1de3..c327419284 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 d7c31edcbf..2a49077899 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()