• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /**
2  * Copyright 2020-2021 Huawei Technologies Co., Ltd
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef MINDSPORE_CORE_OPS_COS_H_
18 #define MINDSPORE_CORE_OPS_COS_H_
19 #include <vector>
20 #include <memory>
21 
22 #include "ops/primitive_c.h"
23 #include "ops/op_utils.h"
24 #include "abstract/abstract_value.h"
25 
26 namespace mindspore {
27 namespace ops {
28 /// \brief Computes cosine of input element-wise. Refer to Python API @ref mindspore.ops.Cos for more details.
29 class MS_CORE_API Cos : public PrimitiveC {
30  public:
31   /// \brief Constructor.
Cos()32   Cos() : PrimitiveC(prim::kPrimCos->name()) {}
33   /// \brief Destructor.
34   ~Cos() = default;
35   MS_DECLARE_PARENT(Cos, PrimitiveC);
36   /// \brief Init. Refer to the parameters of Python API @ref mindspore.ops.Cos for the inputs.
37   void Init(float alpha = 0.0);
38 };
39 }  // namespace ops
40 }  // namespace mindspore
41 #endif  // MINDSPORE_CORE_OPS_COS_H_
42