From eab0fe4e8df9ab72b845284bcd21e24bd637af9c Mon Sep 17 00:00:00 2001 From: jiangjinsheng Date: Thu, 7 May 2020 17:13:16 +0800 Subject: [PATCH] support float32 --- mindspore/ops/_op_impl/tbe/batch_to_space.py | 1 + mindspore/ops/_op_impl/tbe/space_to_batch.py | 1 + 2 files changed, 2 insertions(+) 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()