support float32

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()
 
 

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()