Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!31888 [MS][LITE]fix core ops and lite adapter new mindapi alarm 

Merge pull request !31888 from luoyuan/core-3
This commit is contained in:
i-robot 2022-03-25 10:39:48 +00:00 committed by Gitee
parent 14199a5685 7feb366fe4
commit c9668f23e8
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
100 changed files with 253 additions and 201 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/core/ops/LayerNormBetaGammaBackprop.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,10 +26,12 @@
namespace mindspore {
namespace ops {
constexpr auto kNameLayerNormBetaGammaBackprop = "LayerNormBetaGammaBackprop";
class MIND_API LayerNormBetaGammaBackprop : public BaseOperator {
public:
MIND_API_BASE_MEMBER(LayerNormBetaGammaBackprop);
LayerNormBetaGammaBackprop() : BaseOperator("LayerNormBetaGammaBackprop") {}
LayerNormBetaGammaBackprop() : BaseOperator(kNameLayerNormBetaGammaBackprop) {}
void Init() const {}
};

6
mindspore/core/ops/LayerNormXBackprop.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,10 +26,12 @@
namespace mindspore {
namespace ops {
constexpr auto kNameLayerNormXBackprop = "LayerNormXBackprop";
class MIND_API LayerNormXBackprop : public BaseOperator {
public:
MIND_API_BASE_MEMBER(LayerNormXBackprop);
LayerNormXBackprop() : BaseOperator("LayerNormXBackprop") {}
LayerNormXBackprop() : BaseOperator(kNameLayerNormXBackprop) {}
void Init() const {}
};

2
mindspore/core/ops/abs.cc
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.

6
mindspore/core/ops/abs.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,13 +24,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameAbs = "Abs";
/// \brief Returns absolute value of a tensor element-wise.
/// Refer to Python API @ref mindspore.ops.Abs for more details.
class MIND_API Abs : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Abs);
/// \brief Constructor.
Abs() : BaseOperator("Abs") { InitIOName({"input_x"}, {"output"}); }
Abs() : BaseOperator(kNameAbs) { InitIOName({"input_x"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Abs for the inputs.
void Init() const {}
};

3
mindspore/core/ops/apply_adagrad_d_a.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -42,7 +42,6 @@ class MIND_API ApplyAdagradDA : public BaseOperator {
abstract::AbstractBasePtr ApplyAdagradDAInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);
} // namespace ops
} // namespace mindspore
#endif // MINDSPORE_CORE_OPS_APPLY_ADAGRAD_D_A_H_

6
mindspore/core/ops/approximate_equal.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -23,10 +23,12 @@
namespace mindspore {
namespace ops {
constexpr auto kNameApproximateEqual = "ApproximateEqual";
class MIND_API ApproximateEqual : public BaseOperator {
public:
MIND_API_BASE_MEMBER(ApproximateEqual);
ApproximateEqual() : BaseOperator("ApproximateEqual") {}
ApproximateEqual() : BaseOperator(kNameApproximateEqual) {}
void Init() {}
};
abstract::AbstractBasePtr ApproximateEqualInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,

3
mindspore/core/ops/assign.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -34,7 +34,6 @@ class MIND_API Assign : public BaseOperator {
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Assign for the inputs.
void Init() const {}
};
} // namespace ops
} // namespace mindspore

6
mindspore/core/ops/avg_pool_3d.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,12 +26,14 @@
namespace mindspore {
namespace ops {
constexpr auto kNameAvgPool3D = "AvgPool3D";
/// \brief 3D Average pooling operation. Refer to Python API @ref mindspore.ops.AvgPool3D for more details.
class MIND_API AvgPool3D : public BaseOperator {
public:
MIND_API_BASE_MEMBER(AvgPool3D);
/// \brief Constructor.
AvgPool3D() : BaseOperator("AvgPool3D") { InitIOName({"input"}, {"output"}); }
AvgPool3D() : BaseOperator(kNameAvgPool3D) { InitIOName({"input"}, {"output"}); }
};
abstract::AbstractBasePtr AvgPool3DInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,

6
mindspore/core/ops/batch_matmul.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -23,13 +23,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameBatchMatMul = "BatchMatMul";
/// \brief Computes matrix multiplication between two tensors by batch.
/// Refer to Python API @ref mindspore.ops.BatchMatmul for more details.
class MIND_API BatchMatmul : public BaseOperator {
public:
MIND_API_BASE_MEMBER(BatchMatmul);
/// \brief Constructor.
BatchMatmul() : BaseOperator("BatchMatMul") { InitIOName({"x1", "x2"}, {"output"}); }
BatchMatmul() : BaseOperator(kNameBatchMatMul) { InitIOName({"x1", "x2"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.BatchMatmul for the inputs.
void Init(bool transpose_a = false, bool transpose_b = false);
/// \brief Set transpose_a.

3
mindspore/core/ops/batch_norm.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -68,7 +68,6 @@ class MIND_API BatchNorm : public BaseOperator {
abstract::AbstractBasePtr BatchNormInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);
using PrimBatchNormPtr = std::shared_ptr<BatchNorm>;
} // namespace ops
} // namespace mindspore

5
mindspore/core/ops/bias_add.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -27,12 +27,13 @@
namespace mindspore {
namespace ops {
constexpr auto kNameBiasAdd = "BiasAdd";
/// \brief Returns sum of input and bias tensor. Refer to Python API @ref mindspore.ops.BiasAdd for more details.
class MIND_API BiasAdd : public BaseOperator {
public:
MIND_API_BASE_MEMBER(BiasAdd);
/// \brief Constructor.
BiasAdd() : BaseOperator("BiasAdd") { InitIOName({"x", "b"}, {"output"}); }
BiasAdd() : BaseOperator(kNameBiasAdd) { InitIOName({"x", "b"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.BiasAdd for the inputs.
void Init(const Format &format = NCHW);
/// \brief Set format.

6
mindspore/core/ops/broadcast_to.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,13 +25,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameBroadcastTo = "BroadcastTo";
/// \brief Broadcasts input tensor to a given shape.
/// Refer to Python API @ref mindspore.ops.BroadcastTo for more details.
class MIND_API BroadcastTo : public BaseOperator {
public:
MIND_API_BASE_MEMBER(BroadcastTo);
/// \brief Constructor.
BroadcastTo() : BaseOperator("BroadcastTo") {}
BroadcastTo() : BaseOperator(kNameBroadcastTo) {}
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.BroadcastTo for the inputs.
void Init(const std::vector<int64_t> &shape);
/// \brief Set shape.

4
mindspore/core/ops/celu.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -29,7 +29,7 @@ namespace ops {
constexpr auto kNameCeLU = "CeLU";
/// \brief Computes CeLU (Continuously differentiable exponential linear units) of input tensors element-wise.
/// Refer to Python API @ref mindspore.ops.CeLU for more details.
class CeLU : public BaseOperator {
class MIND_API CeLU : public BaseOperator {
public:
MIND_API_BASE_MEMBER(CeLU);
/// \brief Constructor.

2
mindspore/core/ops/complex.cc
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.

6
mindspore/core/ops/complex.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,13 +24,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameComplex = "Complex";
/// \brief Returns a complex Tensor from the real and imaginary part.
/// Refer to Python API @ref mindspore.ops.Complex for more details.
class MIND_API Complex : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Complex);
/// \brief Constructor.
Complex() : BaseOperator("Square") { InitIOName({"s", "input_imag"}, {"output"}); }
Complex() : BaseOperator(kNameComplex) { InitIOName({"s", "input_imag"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Complex for the inputs.
void Init() {}
};

6
mindspore/core/ops/conj.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,13 +24,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameConj = "Conj";
/// \brief Returns a Tensor that is the conjugate part of the input.
/// Refer to Python API @ref mindspore.ops.Conj for more details.
class MIND_API Conj : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Conj);
/// \brief Constructor.
Conj() : BaseOperator("Conj") { InitIOName({"input"}, {"output"}); }
Conj() : BaseOperator(kNameConj) { InitIOName({"input"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Conj for the inputs.
void Init() {}
};

6
mindspore/core/ops/cos.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -23,12 +23,14 @@
namespace mindspore {
namespace ops {
constexpr auto kNameCos = "Cos";
/// \brief Computes cosine of input element-wise. Refer to Python API @ref mindspore.ops.Cos for more details.
class MIND_API Cos : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Cos);
/// \brief Constructor.
Cos() : BaseOperator("Cos") {}
Cos() : BaseOperator(kNameCos) {}
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Cos for the inputs.
void Init(float alpha = 0.0);
};

4
mindspore/core/ops/cosh.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,7 +24,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameCosh = "Cosh";
class Cosh : public BaseOperator {
class MIND_API Cosh : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Cosh);
Cosh() : BaseOperator(kNameCosh) { InitIOName({"x"}, {"output"}); }

6
mindspore/core/ops/ctcloss.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -23,13 +23,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameCTCLoss = "CTCLoss";
/// \brief Calculates the CTC (Connectionist Temporal Classification) loss and the gradient.
/// Refer to Python API @ref mindspore.ops.CTCLoss for more details.
class MIND_API CTCLoss : public BaseOperator {
public:
MIND_API_BASE_MEMBER(CTCLoss);
/// \brief Constructor.
CTCLoss() : BaseOperator("CTCLoss") {}
CTCLoss() : BaseOperator(kNameCTCLoss) {}
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.CTCLoss for the inputs.
void Init() const {}
};

4
mindspore/core/ops/cummin.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -27,7 +27,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameCummin = "Cummin";
class Cummin : public BaseOperator {
class MIND_API Cummin : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Cummin);
Cummin() : BaseOperator(kNameCummin) { InitIOName({"x"}, {"y"}); }

3
mindspore/core/ops/custom.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -60,7 +60,6 @@ class MIND_API Custom : public BaseOperator {
/// \return a map which contains all attributes of the custom op.
std::map<std::string, std::vector<uint8_t>> get_attr() const;
};
} // namespace ops
} // namespace mindspore

6
mindspore/core/ops/diag.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -23,13 +23,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameDiag = "Diag";
/// \brief Constructs a diagonal tensor with a given diagonal values.
/// Refer to Python API @ref mindspore.ops.Diag for more details.
class MIND_API Diag : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Diag);
/// \brief Constructor.
Diag() : BaseOperator("Diag") { InitIOName({"input_x"}, {"output"}); }
Diag() : BaseOperator(kNameDiag) { InitIOName({"input_x"}, {"output"}); }
};
abstract::AbstractBasePtr DiagInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);

6
mindspore/core/ops/diag_part.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -23,13 +23,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameDiagPart = "DiagPart";
/// \brief Extracts the diagonal part from given tensor.
/// Refer to Python API @ref mindspore.ops.DiagPart for more details.
class MIND_API DiagPart : public BaseOperator {
public:
MIND_API_BASE_MEMBER(DiagPart);
/// \brief Constructor.
DiagPart() : BaseOperator("DiagPart") { InitIOName({"input_x"}, {"output"}); }
DiagPart() : BaseOperator(kNameDiagPart) { InitIOName({"input_x"}, {"output"}); }
};
abstract::AbstractBasePtr DiagPartInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);

4
mindspore/core/ops/div_no_nan.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,7 +26,7 @@ namespace mindspore {
namespace ops {
constexpr auto kNameDivNoNan = "DivNoNan";
class DivNoNan : public BaseOperator {
class MIND_API DivNoNan : public BaseOperator {
public:
MIND_API_BASE_MEMBER(DivNoNan);
DivNoNan() : BaseOperator(kNameDivNoNan) { InitIOName({"x1", "x2"}, {"y"}); }

6
mindspore/core/ops/dropout_do_mask.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,13 +24,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameDropoutDoMask = "DropoutDoMask";
/// \brief Applies dropout mask on the input tensor.
/// Refer to Python API @ref mindspore.ops.DropoutDoMask for more details.
class MIND_API DropoutDoMask : public BaseOperator {
public:
MIND_API_BASE_MEMBER(DropoutDoMask);
/// \brief Constructor.
DropoutDoMask() : BaseOperator("DropoutDoMask") {}
DropoutDoMask() : BaseOperator(kNameDropoutDoMask) {}
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.DropoutDoMask for the inputs.
void Init() const {}
};

6
mindspore/core/ops/dropout_gen_mask.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,13 +24,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameDropoutGenMask = "DropoutGenMask";
/// \brief Generates the mask value for the input shape.
/// Refer to Python API @ref mindspore.ops.DropoutGenMask for more details.
class MIND_API DropoutGenMask : public BaseOperator {
public:
MIND_API_BASE_MEMBER(DropoutGenMask);
/// \brief Constructor.
DropoutGenMask() : BaseOperator("DropoutGenMask") {}
DropoutGenMask() : BaseOperator(kNameDropoutGenMask) {}
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.DropoutGenMask for the inputs.
void Init() const {}
};

6
mindspore/core/ops/dtype.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,13 +25,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameDType = "DType";
/// \brief Returns the data type of the input tensor as mindspore.dtype.
/// Refer to Python API @ref mindspore.ops.DType for more details.
class MIND_API DType : public BaseOperator {
public:
MIND_API_BASE_MEMBER(DType);
/// \brief Constructor.
DType() : BaseOperator("DType") { InitIOName({"x"}, {"output"}); }
DType() : BaseOperator(kNameDType) { InitIOName({"x"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.DType for the inputs.
void Init() const {}
};

8
mindspore/core/ops/dynamic_broadcast_to.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,10 +25,12 @@
namespace mindspore {
namespace ops {
class DynamicBroadcastTo : public BaseOperator {
constexpr auto kNameDynamicBroadcastTo = "DynamicBroadcastTo";
class MIND_API DynamicBroadcastTo : public BaseOperator {
public:
MIND_API_BASE_MEMBER(DynamicBroadcastTo);
DynamicBroadcastTo() : BaseOperator("DynamicBroadcastTo") { InitIOName({"x", "shape"}, {"y"}); }
DynamicBroadcastTo() : BaseOperator(kNameDynamicBroadcastTo) { InitIOName({"x", "shape"}, {"y"}); }
void Init() {}
};

4
mindspore/core/ops/dynamic_resize_nearest_neighbor.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,7 +26,7 @@ namespace mindspore {
namespace ops {
constexpr auto kNameDynamicResizeNearestNeighbor = "DynamicResizeNearestNeighbor";
class DynamicResizeNearestNeighbor : public BaseOperator {
class MIND_API DynamicResizeNearestNeighbor : public BaseOperator {
public:
MIND_API_BASE_MEMBER(DynamicResizeNearestNeighbor);
DynamicResizeNearestNeighbor() : BaseOperator(kNameDynamicResizeNearestNeighbor) {}

4
mindspore/core/ops/equal.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -31,7 +31,7 @@ class MIND_API Equal : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Equal);
/// \brief Constructor.
Equal() : BaseOperator("Equal") { InitIOName({"x", "y"}, {"output"}); }
Equal() : BaseOperator(kNameEqual) { InitIOName({"x", "y"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Equal for the inputs.
void Init() const {}
};

2
mindspore/core/ops/erfc.cc
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.

2
mindspore/core/ops/erfc.h
View File

@ -24,7 +24,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameErfc = "Erfc";
class Erfc : public BaseOperator {
class MIND_API Erfc : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Erfc);
Erfc() : BaseOperator(kNameErfc) { InitIOName({"x"}, {"y"}); }

4
mindspore/core/ops/exp.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -31,7 +31,7 @@ class MIND_API Exp : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Exp);
/// \brief Constructor.
Exp() : BaseOperator("Exp") { InitIOName({"x"}, {"y"}); }
Exp() : BaseOperator(kNameExp) { InitIOName({"x"}, {"y"}); }
explicit Exp(const std::string k_name) : BaseOperator(k_name) { InitIOName({"x"}, {"y"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Exp for the inputs.
void Init() const {}

4
mindspore/core/ops/fast_gelu.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -28,7 +28,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameFastGeLU = "FastGeLU";
class FastGeLU : public BaseOperator {
class MIND_API FastGeLU : public BaseOperator {
public:
MIND_API_BASE_MEMBER(FastGeLU);
FastGeLU() : BaseOperator(kNameFastGeLU) { InitIOName({"x"}, {"y"}); }

3
mindspore/core/ops/floor.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -41,7 +41,6 @@ abstract::AbstractBasePtr FloorInfer(const abstract::AnalysisEnginePtr &, const
const std::vector<abstract::AbstractBasePtr> &input_args);
using PrimFloorPtr = std::shared_ptr<Floor>;
} // namespace ops
} // namespace mindspore

6
mindspore/core/ops/gather_d.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,13 +25,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameGatherD = "GatherD";
/// \brief Gathers values along an axis specified by dimension.
/// Refer to Python API @ref mindspore.ops.GatherD for more details.
class MIND_API GatherD : public BaseOperator {
public:
MIND_API_BASE_MEMBER(GatherD);
/// \brief Constructor.
GatherD() : BaseOperator("GatherD") { InitIOName({"x", "dim", "index"}, {"output"}); }
GatherD() : BaseOperator(kNameGatherD) { InitIOName({"x", "dim", "index"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.GatherD for the inputs.
void Init() const {}
};

3
mindspore/core/ops/gelu.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -33,7 +33,6 @@ class MIND_API GeLU : public BaseOperator {
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.GeLU for the inputs.
void Init() const {}
};
} // namespace ops
} // namespace mindspore
#endif // MINDSPORE_CORE_OPS_GELU_H_

4
mindspore/core/ops/getnext.h
View File

@ -1,6 +1,6 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -33,7 +33,7 @@ class MIND_API GetNext : public BaseOperator {
public:
MIND_API_BASE_MEMBER(GetNext);
/// \brief Constructor.
GetNext() : BaseOperator("GetNext") {}
GetNext() : BaseOperator(kNameGetNext) {}
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.GetNext for the inputs.
void Init() const {}
};

3
mindspore/core/ops/glu.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -32,7 +32,6 @@ class MIND_API GLU : public BaseOperator {
void set_axis(int64_t axis);
int64_t get_axis() const;
};
} // namespace ops
} // namespace mindspore
#endif // MINDSPORE_CORE_OPS_GLU_H_

6
mindspore/core/ops/grad/avg_pool_3d_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,10 +26,12 @@
namespace mindspore {
namespace ops {
constexpr auto kNameAvgPool3DGrad = "AvgPool3DGrad";
class MIND_API AvgPool3DGrad : public BaseOperator {
public:
MIND_API_BASE_MEMBER(AvgPool3DGrad);
AvgPool3DGrad() : BaseOperator("AvgPool3DGrad") { InitIOName({"origin_input_size", "grad"}, {"output"}); }
AvgPool3DGrad() : BaseOperator(kNameAvgPool3DGrad) { InitIOName({"origin_input_size", "grad"}, {"output"}); }
};
abstract::AbstractBasePtr AvgPool3DGradInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,

6
mindspore/core/ops/grad/dropout_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,10 +24,12 @@
namespace mindspore {
namespace ops {
constexpr auto kNameDropoutGrad = "DropoutGrad";
class MIND_API DropoutGrad : public BaseOperator {
public:
MIND_API_BASE_MEMBER(DropoutGrad);
DropoutGrad() : BaseOperator("DropoutGrad") {}
DropoutGrad() : BaseOperator(kNameDropoutGrad) {}
void Init(const float keep_prob = 0.5);
void set_keep_prob(const float keep_prob);
float get_keep_prob() const;

7
mindspore/core/ops/grad/fast_gelu_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,10 +26,11 @@
namespace mindspore {
namespace ops {
constexpr auto kNameFastGeLUGrad = "FastGeLUGrad";
class FastGeLUGrad : public BaseOperator {
class MIND_API FastGeLUGrad : public BaseOperator {
public:
MIND_API_BASE_MEMBER(FastGeLUGrad);
FastGeLUGrad() : BaseOperator("FastGeLUGrad") { InitIOName({"x"}, {"output"}); }
FastGeLUGrad() : BaseOperator(kNameFastGeLUGrad) { InitIOName({"x"}, {"output"}); }
};
} // namespace ops
} // namespace mindspore

4
mindspore/core/ops/grad/gelu_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,7 +25,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameGeLUGrad = "GeLUGrad";
class GeLUGrad : public BaseOperator {
class MIND_API GeLUGrad : public BaseOperator {
public:
MIND_API_BASE_MEMBER(GeLUGrad);
GeLUGrad() : BaseOperator(kNameGeLUGrad) { InitIOName({"dy", "x", "y"}, {"z"}); }

4
mindspore/core/ops/grad/inv_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,7 +25,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameInvGrad = "InvGrad";
class InvGrad : public BaseOperator {
class MIND_API InvGrad : public BaseOperator {
public:
MIND_API_BASE_MEMBER(InvGrad);
InvGrad() : BaseOperator(kNameInvGrad) { InitIOName({"x", "grad"}, {"y"}); }

3
mindspore/core/ops/grad/mul_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -29,7 +29,6 @@ class MIND_API MulGrad : public BaseOperator {
MulGrad() : BaseOperator(kNameMulGrad) {}
void Init() const {}
};
} // namespace ops
} // namespace mindspore

4
mindspore/core/ops/grad/resize_nearest_neighbor_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,7 +26,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameResizeNearestNeighborGrad = "ResizeNearestNeighborGrad";
class ResizeNearestNeighborGrad : public BaseOperator {
class MIND_API ResizeNearestNeighborGrad : public BaseOperator {
public:
MIND_API_BASE_MEMBER(ResizeNearestNeighborGrad);
ResizeNearestNeighborGrad() : BaseOperator(kNameResizeNearestNeighborGrad) {}

6
mindspore/core/ops/grad/strided_slice_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,10 +25,12 @@
namespace mindspore {
namespace ops {
constexpr auto kNameStridedSliceGrad = "StridedSliceGrad";
class MIND_API StridedSliceGrad : public BaseOperator {
public:
MIND_API_BASE_MEMBER(StridedSliceGrad);
StridedSliceGrad() : BaseOperator("StridedSliceGrad") {
StridedSliceGrad() : BaseOperator(kNameStridedSliceGrad) {
InitIOName({"dy", "shapex", "begin", "end", "strides"}, {"output"});
}

3
mindspore/core/ops/grad/sub_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -29,7 +29,6 @@ class MIND_API SubGrad : public BaseOperator {
SubGrad() : BaseOperator(kNameSubGrad) {}
void Init() const {}
};
} // namespace ops
} // namespace mindspore

4
mindspore/core/ops/grad/tanh_grad.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -27,7 +27,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameTanhGrad = "TanhGrad";
class TanhGrad : public BaseOperator {
class MIND_API TanhGrad : public BaseOperator {
public:
MIND_API_BASE_MEMBER(TanhGrad);
/// \brief Constructor.

3
mindspore/core/ops/gru.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -52,7 +52,6 @@ class MIND_API GRU : public BaseOperator {
/// \return a boolean value.
bool get_bidirectional() const;
};
} // namespace ops
} // namespace mindspore

3
mindspore/core/ops/hsigmoid.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -35,7 +35,6 @@ class MIND_API HSigmoid : public BaseOperator {
abstract::AbstractBasePtr HSigmoidInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);
} // namespace ops
} // namespace mindspore

6
mindspore/core/ops/imag.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,13 +24,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameImag = "Imag";
/// \brief Returns a Tensor that is the imag part of the input.
/// Refer to Python API @ref mindspore.ops.Imag for more details.
class MIND_API Imag : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Imag);
/// \brief Constructor.
Imag() : BaseOperator("Imag") { InitIOName({"input"}, {"output"}); }
Imag() : BaseOperator(kNameImag) { InitIOName({"input"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Imag for the inputs.
void Init() {}
};

2
mindspore/core/ops/index_add.cc
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.

5
mindspore/core/ops/index_add.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -29,7 +29,7 @@ namespace ops {
constexpr auto kNameIndexAdd = "IndexAdd";
/// \brief Adds tensor y to specified axis and indices of tensor x.
/// Refer to Python API @ref mindspore.ops.IndexAdd for more details.
class IndexAdd : public BaseOperator {
class MIND_API IndexAdd : public BaseOperator {
public:
MIND_API_BASE_MEMBER(IndexAdd);
/// \brief Constructor.
@ -38,7 +38,6 @@ class IndexAdd : public BaseOperator {
abstract::AbstractBasePtr IndexAddInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);
} // namespace ops
} // namespace mindspore

4
mindspore/core/ops/inv.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,7 +24,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameInv = "Inv";
class Inv : public BaseOperator {
class MIND_API Inv : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Inv);
Inv() : BaseOperator(kNameInv) { InitIOName({"x"}, {"y"}); }

3
mindspore/core/ops/invert_permutation.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -32,7 +32,6 @@ class MIND_API InvertPermutation : public BaseOperator {
/// \brief Constructor.
InvertPermutation() : BaseOperator(kNameInvertPermutation) {}
};
} // namespace ops
} // namespace mindspore

6
mindspore/core/ops/iou.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,10 +24,12 @@
namespace mindspore {
namespace ops {
constexpr auto kNameIOU = "IOU";
class MIND_API IOU : public BaseOperator {
public:
MIND_API_BASE_MEMBER(IOU);
IOU() : BaseOperator("IOU") { InitIOName({"x,y"}, {"output"}); }
IOU() : BaseOperator(kNameIOU) { InitIOName({"x,y"}, {"output"}); }
void Init() {}
};
} // namespace ops

4
mindspore/core/ops/is_close.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -27,7 +27,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameIsClose = "IsClose";
class IsClose : public BaseOperator {
class MIND_API IsClose : public BaseOperator {
public:
MIND_API_BASE_MEMBER(IsClose);
IsClose() : BaseOperator(kNameIsClose) { InitIOName({"x1", "x2"}, {"y"}); }

4
mindspore/core/ops/is_nan.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,7 +25,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameIsNan = "IsNan";
class IsNan : public BaseOperator {
class MIND_API IsNan : public BaseOperator {
public:
MIND_API_BASE_MEMBER(IsNan);
IsNan() : BaseOperator(kNameIsNan) { InitIOName({"x"}, {"y"}); }

3
mindspore/core/ops/layer_norm.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -59,7 +59,6 @@ class MIND_API LayerNorm : public BaseOperator {
abstract::AbstractBasePtr LayerNormInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);
using PrimLayerNormPtr = std::shared_ptr<LayerNorm>;
} // namespace ops
} // namespace mindspore

8
mindspore/core/ops/layer_norm_beta_gamma_backprop_v2.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,10 +25,12 @@
namespace mindspore {
namespace ops {
class LayerNormBetaGammaBackpropV2 : public BaseOperator {
constexpr auto kNameLayerNormBetaGammaBackpropV2 = "LayerNormBetaGammaBackpropV2";
class MIND_API LayerNormBetaGammaBackpropV2 : public BaseOperator {
public:
MIND_API_BASE_MEMBER(LayerNormBetaGammaBackpropV2);
LayerNormBetaGammaBackpropV2() : BaseOperator("LayerNormBetaGammaBackpropV2") {}
LayerNormBetaGammaBackpropV2() : BaseOperator(kNameLayerNormBetaGammaBackpropV2) {}
void Init(const std::vector<int64_t> &shape_gamma);
void set_shape_gamma(const std::vector<int64_t> &shape_gamma);
std::vector<int64_t> get_shape_gamma() const;

4
mindspore/core/ops/log.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -30,7 +30,7 @@ class MIND_API Log : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Log);
/// \brief Constructor.
Log() : BaseOperator("Log") { InitIOName({"x"}, {"y"}); }
Log() : BaseOperator(kNameLog) { InitIOName({"x"}, {"y"}); }
};
abstract::AbstractBasePtr LogInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);

6
mindspore/core/ops/log1p.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,13 +25,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameLog1p = "Log1p";
/// \brief Returns the natural logarithm of one plus the input tensor element-wise.
/// Refer to Python API @ref mindspore.ops.Log1p for more details.
class MIND_API Log1p : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Log1p);
/// \brief Constructor.
Log1p() : BaseOperator("Log1p") { InitIOName({"x"}, {"y"}); }
Log1p() : BaseOperator(kNameLog1p) { InitIOName({"x"}, {"y"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Log1p for the inputs.
void Init() const {}
};

4
mindspore/core/ops/lp_norm.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,7 +25,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameLpNorm = "LpNorm";
class LpNorm : public BaseOperator {
class MIND_API LpNorm : public BaseOperator {
public:
MIND_API_BASE_MEMBER(LpNorm);
LpNorm() : BaseOperator(kNameLpNorm) { InitIOName({"input"}, {"output"}); }

3
mindspore/core/ops/lp_normalization.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -57,7 +57,6 @@ class MIND_API LpNormalization : public BaseOperator {
/// \return the norm series.
int64_t get_p() const;
};
} // namespace ops
} // namespace mindspore

4
mindspore/core/ops/masked_fill.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -27,7 +27,7 @@ namespace ops {
constexpr auto kNameMaskedFill = "MaskedFill";
/// \brief Fills elements of self tensor with value where mask is True.
/// Refer to Python API @ref mindspore.ops.MaskedFill for more details.
class MaskedFill : public BaseOperator {
class MIND_API MaskedFill : public BaseOperator {
public:
MIND_API_BASE_MEMBER(MaskedFill);
/// \brief Constructor.

8
mindspore/core/ops/matrix_diag_part.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,14 +24,14 @@
namespace mindspore {
namespace ops {
constexpr auto kNameMatrixDiagPart = "MatrixDiagPartV3";
constexpr auto kNameMatrixDiagPartV3 = "MatrixDiagPartV3";
/// \brief get the specified part of the inner most diag matrix of a matrix, fill with padding value .
/// Refer to Python API @ref mindspore.ops.MatrixDiagPart for more details.
class MatrixDiagPartV3 : public BaseOperator {
class MIND_API MatrixDiagPartV3 : public BaseOperator {
public:
MIND_API_BASE_MEMBER(MatrixDiagPartV3);
/// \brief Constructor.
MatrixDiagPartV3() : BaseOperator(kNameMatrixDiagPart) { InitIOName({"input", "k", "padding_value"}, {"output"}); }
MatrixDiagPartV3() : BaseOperator(kNameMatrixDiagPartV3) { InitIOName({"input", "k", "padding_value"}, {"output"}); }
};
abstract::AbstractBasePtr MatrixDiagPartInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,

2
mindspore/core/ops/mulnonan.h
View File

@ -26,7 +26,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameMulNoNan = "MulNoNan";
class MulNoNan : public BaseOperator {
class MIND_API MulNoNan : public BaseOperator {
public:
MIND_API_BASE_MEMBER(MulNoNan);
MulNoNan() : BaseOperator(kNameMulNoNan) { InitIOName({"x", "y"}, {"output"}); }

4
mindspore/core/ops/neg.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -30,7 +30,7 @@ class MIND_API Neg : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Neg);
/// \brief Constructor.
Neg() : BaseOperator("Neg") { InitIOName({"x"}, {"y"}); }
Neg() : BaseOperator(kNameNeg) { InitIOName({"x"}, {"y"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Neg for the inputs.
void Init() const {}
};

4
mindspore/core/ops/non_max_suppression_v3.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -29,7 +29,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameNonMaxSuppressionV3 = "NonMaxSuppressionV3";
class NonMaxSuppressionV3 : public BaseOperator {
class MIND_API NonMaxSuppressionV3 : public BaseOperator {
public:
MIND_API_BASE_MEMBER(NonMaxSuppressionV3);
NonMaxSuppressionV3() : BaseOperator(kNameNonMaxSuppressionV3) {

4
mindspore/core/ops/not_equal.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -31,7 +31,7 @@ class MIND_API NotEqual : public BaseOperator {
public:
MIND_API_BASE_MEMBER(NotEqual);
/// \brief Constructor.
NotEqual() : BaseOperator("NotEqual") { InitIOName({"x", "y"}, {"output"}); }
NotEqual() : BaseOperator(kNameNotEqual) { InitIOName({"x", "y"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.NotEqual for the inputs.
void Init() const {}
};

6
mindspore/core/ops/one_hot.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,12 +24,14 @@
namespace mindspore {
namespace ops {
constexpr auto kNameOneHot = "OneHot";
/// \brief Computes a one-hot tensor. Refer to Python API @ref mindspore.ops.OneHot for more details.
class MIND_API OneHot : public BaseOperator {
public:
MIND_API_BASE_MEMBER(OneHot);
/// \brief Constructor.
OneHot() : BaseOperator("OneHot") { InitIOName({"indices", "depth", "on_value", "off_value"}, {"output"}); }
OneHot() : BaseOperator(kNameOneHot) { InitIOName({"indices", "depth", "on_value", "off_value"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.OneHot for the inputs.
void Init(const int64_t axis);
/// \brief Set axis.

8
mindspore/core/ops/ones.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,12 +26,14 @@
namespace mindspore {
namespace ops {
constexpr auto kNameOnes = "Ones";
/// \brief Creates a tensor filled with value ones. Refer to Python API @ref mindspore.ops.Ones for more details.
class Ones : public BaseOperator {
class MIND_API Ones : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Ones);
/// \brief Constructor.
Ones() : BaseOperator("Ones") {}
Ones() : BaseOperator(kNameOnes) {}
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Ones for the inputs.
void Init() const {}
};

3
mindspore/core/ops/ragged_range.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -34,7 +34,6 @@ class MIND_API RaggedRange : public BaseOperator {
/// \brief Method to init the op.
void Init() const {}
};
} // namespace ops
} // namespace mindspore

6
mindspore/core/ops/real.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,13 +24,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameReal = "Real";
/// \brief Returns a Tensor that is the real part of the input.
/// Refer to Python API @ref mindspore.ops.Real for more details.
class MIND_API Real : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Real);
/// \brief Constructor.
Real() : BaseOperator("Real") { InitIOName({"input"}, {"output"}); }
Real() : BaseOperator(kNameReal) { InitIOName({"input"}, {"output"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Real for the inputs.
void Init() {}
};

4
mindspore/core/ops/reciprocal.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -31,7 +31,7 @@ class MIND_API Reciprocal : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Reciprocal);
/// \brief Constructor.
Reciprocal() : BaseOperator("Reciprocal") { InitIOName({"x"}, {"y"}); }
Reciprocal() : BaseOperator(kNameReciprocal) { InitIOName({"x"}, {"y"}); }
/// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Reciprocal for the inputs.
void Init() const {}
};

4
mindspore/core/ops/reluv2.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -32,7 +32,7 @@ class MIND_API ReLUV2 : public BaseOperator {
public:
MIND_API_BASE_MEMBER(ReLUV2);
/// \brief Constructor.
ReLUV2() : BaseOperator("ReluV2") { InitIOName({"x"}, {"output", "mask"}); }
ReLUV2() : BaseOperator(kNameReLUV2) { InitIOName({"x"}, {"output", "mask"}); }
/// \brief Constructor.
explicit ReLUV2(const std::string k_name) : BaseOperator(k_name) { InitIOName({"x"}, {"output", "mask"}); }
/// \brief Init.

3
mindspore/core/ops/resize_nearest_neighbor.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -47,7 +47,6 @@ class MIND_API ResizeNearestNeighbor : public BaseOperator {
/// \return align_corners.
bool get_align_corners() const;
};
} // namespace ops
} // namespace mindspore

4
mindspore/core/ops/rint.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,7 +25,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameRint = "Rint";
class Rint : public BaseOperator {
class MIND_API Rint : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Rint);
Rint() : BaseOperator(kNameRint) { InitIOName({"x"}, {"output"}); }

4
mindspore/core/ops/roll.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,7 +26,7 @@ namespace mindspore {
namespace ops {
constexpr auto kNameRoll = "Roll";
/// \brief Rolls the elements of a tensor along an axis.
class Roll : public BaseOperator {
class MIND_API Roll : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Roll);
/// \brief Constructor.

6
mindspore/core/ops/scalar_summary.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,13 +26,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameScalarSummary = "ScalarSummary";
/// \brief Outputs a scalar to a protocol buffer through a scalar summary operator.
/// Refer to Python API @ref mindspore.ops.ScalarSummary for more details.
class MIND_API ScalarSummary : public BaseOperator {
public:
MIND_API_BASE_MEMBER(ScalarSummary);
/// \brief Constructor.
ScalarSummary() : BaseOperator("ScalarSummary") {}
ScalarSummary() : BaseOperator(kNameScalarSummary) {}
/// \brief Init.
void Init();
/// \brief Set side_effect_io.

6
mindspore/core/ops/shape.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,13 +26,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameShape = "Shape";
/// \brief Returns the shape of the input tensor.
/// Refer to Python API @ref mindspore.ops.Shape for more details.
class MIND_API Shape : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Shape);
/// \brief Constructor.
Shape() : BaseOperator("Shape") {}
Shape() : BaseOperator(kNameShape) {}
/// \brief Init.
void Init() const {}
};

4
mindspore/core/ops/sign.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,7 +24,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameSign = "Sign";
class Sign : public BaseOperator {
class MIND_API Sign : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Sign);
Sign() : BaseOperator(kNameSign) { InitIOName({"x"}, {"y"}); }

4
mindspore/core/ops/sinh.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,7 +25,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameSinh = "Sinh";
class Sinh : public BaseOperator {
class MIND_API Sinh : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Sinh);
Sinh() : BaseOperator(kNameSinh) { InitIOName({"x"}, {"output"}); }

3
mindspore/core/ops/size.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -32,7 +32,6 @@ class MIND_API Size : public BaseOperator {
/// \brief Constructor.
Size() : BaseOperator(kNameSize) {}
};
} // namespace ops
} // namespace mindspore

3
mindspore/core/ops/soft_shrink.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -37,7 +37,6 @@ class MIND_API SoftShrink : public BaseOperator {
abstract::AbstractBasePtr SoftShrinkInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
const std::vector<abstract::AbstractBasePtr> &input_args);
} // namespace ops
} // namespace mindspore
#endif // MINDSPORE_CORE_OPS_SOFTSHRINK_H_

4
mindspore/core/ops/sparse_apply_adadelta.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -28,7 +28,7 @@
namespace mindspore {
namespace ops {
constexpr auto kNameSparseApplyAdadelta = "SparseApplyAdadelta";
class SparseApplyAdadelta : public BaseOperator {
class MIND_API SparseApplyAdadelta : public BaseOperator {
public:
MIND_API_BASE_MEMBER(SparseApplyAdadelta);
SparseApplyAdadelta() : BaseOperator(kNameSparseApplyAdadelta) {

4
mindspore/core/ops/sparse_apply_r_m_s_prop.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -29,7 +29,7 @@ namespace mindspore {
namespace ops {
constexpr auto kNameSparseApplyRMSProp = "SparseApplyRMSProp";
/// \brief Update relevant entries according to the rmsprop algorithm.
class SparseApplyRMSProp : public BaseOperator {
class MIND_API SparseApplyRMSProp : public BaseOperator {
public:
MIND_API_BASE_MEMBER(SparseApplyRMSProp);
/// \brief Constructor.

4
mindspore/core/ops/split_v.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -27,7 +27,7 @@ namespace ops {
constexpr auto kNameSplitV = "SplitV";
/// \brief Splits the input tensor into num_split tensors along the given dimension.
/// Refer to Python API @ref mindspore.ops.SplitV for more details.
class SplitV : public BaseOperator {
class MIND_API SplitV : public BaseOperator {
public:
MIND_API_BASE_MEMBER(SplitV);
/// \brief Constructor.

6
mindspore/core/ops/square.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -25,12 +25,14 @@
namespace mindspore {
namespace ops {
constexpr auto kNameSquare = "Square";
/// \brief Returns square of a tensor element-wise. Refer to Python API @ref mindspore.ops.Square for more details.
class MIND_API Square : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Square);
/// \brief Constructor.
Square() : BaseOperator("Square") { InitIOName({"input_x"}, {"output"}); }
Square() : BaseOperator(kNameSquare) { InitIOName({"input_x"}, {"output"}); }
/// \brief Init.
void Init() const {}
};

5
mindspore/core/ops/strided_slice.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -27,12 +27,13 @@
namespace mindspore {
namespace ops {
constexpr auto kNameStridedSlice = "StridedSlice";
/// \brief Extracts a strided slice of a tensor. Refer to Python API @ref mindspore.ops.StridedSlice for more details.
class MIND_API StridedSlice : public BaseOperator {
public:
MIND_API_BASE_MEMBER(StridedSlice);
/// \brief Constructor.
StridedSlice() : BaseOperator("StridedSlice") { InitIOName({"x", "begin", "end", "strides"}, {"output"}); }
StridedSlice() : BaseOperator(kNameStridedSlice) { InitIOName({"x", "begin", "end", "strides"}, {"output"}); }
/// \brief Init. Refer to the parameters of python API @ref mindspore.ops.StridedSlice for the inputs.
void Init(int64_t begin_mask = 0, int64_t end_mask = 0, int64_t ellipsis_mask = 0, int64_t new_axis_mask = 0,
int64_t shrink_axis_mask = 0);

6
mindspore/core/ops/tensor_summary.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,13 +26,15 @@
namespace mindspore {
namespace ops {
constexpr auto kNameTensorSummary = "TensorSummary";
/// \brief Outputs a tensor to a protocol buffer through a tensor summary operator.
/// Refer to Python API @ref mindspore.ops.TensorSummary for more details.
class MIND_API TensorSummary : public BaseOperator {
public:
MIND_API_BASE_MEMBER(TensorSummary);
/// \brief Constructor.
TensorSummary() : BaseOperator("TensorSummary") {}
TensorSummary() : BaseOperator(kNameTensorSummary) {}
/// \brief Init.
void Init();
/// \brief Set side_effect_io.

4
mindspore/core/ops/trunc.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -28,7 +28,7 @@ namespace mindspore {
namespace ops {
constexpr auto kNameTrunc = "Trunc";
/// \brief Returns a new tensor with the truncated integer values of the elements of input.
class Trunc : public BaseOperator {
class MIND_API Trunc : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Trunc);
/// \brief Constructor.

4
mindspore/core/ops/truncate_mod.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,7 +24,7 @@ namespace mindspore {
namespace ops {
constexpr auto kNameTruncateMod = "TruncateMod";
class TruncateMod : public BaseOperator {
class MIND_API TruncateMod : public BaseOperator {
public:
MIND_API_BASE_MEMBER(TruncateMod);
TruncateMod() : BaseOperator(kNameTruncateMod) { InitIOName({"x", "y"}, {"output"}); }

7
mindspore/core/ops/xdivy.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -29,10 +29,11 @@
namespace mindspore {
namespace ops {
constexpr auto kNameXdivy = "Xdivy";
class Xdivy : public BaseOperator {
class MIND_API Xdivy : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Xdivy);
Xdivy() : BaseOperator("Xdivy") { InitIOName({"x", "y"}, {"output"}); }
Xdivy() : BaseOperator(kNameXdivy) { InitIOName({"x", "y"}, {"output"}); }
};
using PrimXdivyPtr = std::shared_ptr<Xdivy>;

6
mindspore/core/ops/zeros.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,12 +26,14 @@
namespace mindspore {
namespace ops {
constexpr auto kNameZeros = "Zeros";
/// \brief Creates a tensor filled with value zeros. Refer to Python API @ref mindspore.ops.Zeros for more details.
class MIND_API Zeros : public BaseOperator {
public:
MIND_API_BASE_MEMBER(Zeros);
/// \brief Constructor.
Zeros() : BaseOperator("Zeros") {}
Zeros() : BaseOperator(kNameZeros) {}
/// \brief Init.
void Init() const {}
};

6
mindspore/core/ops/zeros_like.h
View File

@ -1,5 +1,5 @@
/**
* Copyright 2020-2021 Huawei Technologies Co., Ltd
* Copyright 2020-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -24,12 +24,14 @@
namespace mindspore {
namespace ops {
constexpr auto kNameZerosLike = "ZerosLike";
/// \brief Creates a new tensor. Refer to Python API @ref mindspore.ops.ZerosLike for more details.
class MIND_API ZerosLike : public BaseOperator {
public:
MIND_API_BASE_MEMBER(ZerosLike);
/// \brief Constructor.
ZerosLike() : BaseOperator("ZerosLike") { InitIOName({"x"}, {"y"}); }
ZerosLike() : BaseOperator(kNameZerosLike) { InitIOName({"x"}, {"y"}); }
/// \brief Init.
void Init() const {}
};

5
mindspore/lite/test/config_level0/models_onnx_fp16.cfg
View File

@ -131,9 +131,10 @@ ml_video_edit_moon_mode_sky_refine.onnx;2;1,256,256,4:1,88,88,4 2
ml_video_edit_face_edit_pix2pixHD_unet.onnx 0.5
ml_video_edit_styleCode_part1.onnx 14
ml_video_edit_styleCode_part2.onnx;9 8
poisson_galleryc_c_image.onnx;1:input;1,224,224,3 15
# cur acc for poisson_galleryc_c_text_cast_fp_with_norm_no_flatten.onnx is 14.2%
poisson_galleryc_c_text_cast_fp_with_norm_no_flatten.onnx;1:input 15
poisson_galleryc_c_image.onnx;1:input;1,224,224,3 15
# cur acc for poisson_galleryc_c_text_cast_fp_with_norm_no_flatten.onnx is 4.8%
poisson_galleryc_c_text_cast_fp_with_norm_no_flatten.onnx;1:input 5
ml_video_edit_face_cutout_portraitSeg_1.onnx;2:a,b 2
ml_video_edit_face_cutout_portraitSeg_deconv.onnx;2:a,b 4
ml_video_edit_face_edit_retina_0906.onnx;1:input;1,107,80,3;;fp16_weight 2.5

1
mindspore/lite/tools/converter/adapter/dpico/parser/detection_output_param_helper.cc
View File

@ -242,6 +242,5 @@ int GetDetectionOutputParamFromAttrs(std::vector<mapper::DetectionOutputParam> *
}
return RET_OK;
}
} // namespace dpico
} // namespace mindspore

3
mindspore/lite/tools/optimizer/fusion/fullconnected_fusion.cc
View File

@ -1,5 +1,5 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
* Copyright 2021-2022 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -26,7 +26,6 @@
#include "ops/op_utils.h"
namespace mindspore::opt {
namespace {
constexpr size_t kFcWeightIndex = 2;
constexpr size_t kFcParameterDims = 2;