Program Listing for File eltwise.h

↰ Return to documentation for file (include/converter/include/ops/eltwise.h)

#ifndef MINDSPORE_CORE_OPS_ELTWISE_H_
#define MINDSPORE_CORE_OPS_ELTWISE_H_
#include "mindapi/base/types.h"
#include "ops/base_operator.h"

namespace mindspore {
namespace ops {
constexpr auto kNameEltwise = "Eltwise";
class MIND_API Eltwise : public BaseOperator {
 public:
  MIND_API_BASE_MEMBER(Eltwise);
  Eltwise() : BaseOperator(kNameEltwise) {}

  void Init(const EltwiseMode &mode);

  void set_mode(const EltwiseMode &mode);

  EltwiseMode get_mode() const;
};
}  // namespace ops
}  // namespace mindspore

#endif  // MINDSPORE_CORE_OPS_ELTWISE_H_