Program Listing for File pad_fusion.h

↰ Return to documentation for file (include/converter/include/infer/cxx_api/pad_fusion.h)

#ifndef MINDSPORE_CORE_OPS_PAD_FUSION_H_
#define MINDSPORE_CORE_OPS_PAD_FUSION_H_
#include "mindapi/base/types.h"
#include "infer/pad.h"

namespace mindspore {
namespace ops {
constexpr auto kNamePadFusion = "PadFusion";
class OPS_API PadFusion : public Pad {
 public:
  MIND_API_BASE_MEMBER(PadFusion);
  PadFusion() : Pad(kNamePadFusion) { InitIOName({"x"}, {"y"}); }

  void Init(const PaddingMode &padding_mode, const float constant_value);

  void set_padding_mode(const PaddingMode &padding_mode);

  void set_constant_value(const float constant_value);

  PaddingMode get_padding_mode() const;

  float get_constant_value() const;
};
}  // namespace ops
}  // namespace mindspore

#endif  // MINDSPORE_CORE_OPS_PAD_FUSION_H_