From d4484b569cb4427c65341483742781d5c14f4744 Mon Sep 17 00:00:00 2001 From: VectorSL Date: Thu, 27 Aug 2020 16:05:22 +0800 Subject: [PATCH] FusedBatchNormEx add signature --- mindspore/ops/operations/nn_ops.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index 5e4ee5691b..b8708f1894 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -678,6 +678,13 @@ class FusedBatchNormEx(PrimitiveWithInfer): >>> op = P.FusedBatchNormEx() >>> output = op(input_x, scale, bias, mean, variance) """ + __mindspore_signature__ = ( + ('input_x', sig_rw.RW_READ, sig_kind.KIND_POSITIONAL_KEYWORD, sig_kind.KIND_EMPTY_DEFAULT_VALUE, sig_dtype.T2), + ('scale', sig_rw.RW_WRITE, sig_kind.KIND_POSITIONAL_KEYWORD, sig_kind.KIND_EMPTY_DEFAULT_VALUE, sig_dtype.T), + ('bias', sig_rw.RW_WRITE, sig_kind.KIND_POSITIONAL_KEYWORD, sig_kind.KIND_EMPTY_DEFAULT_VALUE, sig_dtype.T), + ('mean', sig_rw.RW_WRITE, sig_kind.KIND_POSITIONAL_KEYWORD, sig_kind.KIND_EMPTY_DEFAULT_VALUE, sig_dtype.T), + ('variance', sig_rw.RW_WRITE, sig_kind.KIND_POSITIONAL_KEYWORD, sig_kind.KIND_EMPTY_DEFAULT_VALUE, sig_dtype.T), + ) @prim_attr_register def __init__(self, mode=0, epsilon=1e-5, momentum=0.1):