From 2091c80f7db14375b734d10adb6aceb65ee1e4f1 Mon Sep 17 00:00:00 2001 From: wangshuide2020 Date: Mon, 10 May 2021 11:25:16 +0800 Subject: [PATCH] add GPU platform for Mod, SpaceToDepth and DepthToSpace operators. --- mindspore/ops/operations/array_ops.py | 4 ++-- mindspore/ops/operations/math_ops.py | 5 +++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index df91c1500d5..13849b937d7 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -4087,7 +4087,7 @@ class SpaceToDepth(PrimitiveWithInfer): ValueError: If length of shape of `x` is not equal to 4. Supported Platforms: - ``Ascend`` + ``Ascend`` ``GPU`` Examples: >>> x = Tensor(np.random.rand(1,3,2,2), mindspore.float32) @@ -4154,7 +4154,7 @@ class DepthToSpace(PrimitiveWithInfer): ValueError: If length of shape of `x` is not equal to 4. Supported Platforms: - ``Ascend`` + ``Ascend`` ``GPU`` Examples: >>> x = Tensor(np.random.rand(1, 12, 1, 1), mindspore.float32) diff --git a/mindspore/ops/operations/math_ops.py b/mindspore/ops/operations/math_ops.py index 45cfd7cd3aa..7f493e96641 100644 --- a/mindspore/ops/operations/math_ops.py +++ b/mindspore/ops/operations/math_ops.py @@ -2421,7 +2421,7 @@ class Mod(_MathBinaryOp): ValueError: When `input_x` and `input_y` are not the same dtype. Supported Platforms: - ``Ascend`` ``CPU`` + ``Ascend`` ``GPU`` ``CPU`` Examples: >>> input_x = Tensor(np.array([-4.0, 5.0, 6.0]), mindspore.float32) @@ -2482,7 +2482,8 @@ class Floor(PrimitiveWithInfer): class FloorMod(_MathBinaryOp): """ - Computes the remainder of division element-wise. + Computes the remainder of division element-wise. It's a flooring divide. + E.g. :math:`floor(x / y) * y + mod(x, y) = x`. Inputs of `input_x` and `input_y` comply with the implicit type conversion rules to make the data types consistent. The inputs must be two tensors or one tensor and one scalar.