Skip to content

Commit

Permalink
Misprints
Browse files Browse the repository at this point in the history
  • Loading branch information
jane-intel committed Nov 15, 2023
1 parent 48b2f38 commit 9bf85d3
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,9 @@ class TRANSFORMATIONS_API RPE : public ov::op::Op {
OPENVINO_OP("RPE", "ie_internal_opset", op::Op);

RPE() = default;
RPE(const Output<Node>& data, const Output<Node>& sin, const Output<Node>& cos, const int64_t axis);
RPE(const Output<Node>& data, const Output<Node>& sin, const Output<Node>& cos, int64_t axis);

void set_axis(const int64_t axis);
void set_axis(int64_t axis);
int64_t get_axis() const;

bool visit_attributes(AttributeVisitor& visitor) override;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,13 @@ namespace ov {
namespace op {
namespace internal {

RPE::RPE(const Output<Node>& data, const Output<Node>& sin, const Output<Node>& cos, const int64_t axis)
RPE::RPE(const Output<Node>& data, const Output<Node>& sin, const Output<Node>& cos, int64_t axis)
: Op({data, sin, cos}),
m_axis{axis} {
constructor_validate_and_infer_types();
}

void RPE::set_axis(const int64_t axis) {
void RPE::set_axis(int64_t axis) {
m_axis = axis;
}

Expand All @@ -37,7 +37,7 @@ bool RPE::visit_attributes(ov::AttributeVisitor& visitor) {

std::shared_ptr<ov::Node> RPE::clone_with_new_inputs(const ov::OutputVector& new_args) const {
INTERNAL_OP_SCOPE(internal_RoPE_clone_with_new_inputs);
return make_shared<RPE>(new_args.at(0), new_args.at(1), new_args.at(2), m_axis);
return std::make_shared<RPE>(new_args.at(0), new_args.at(1), new_args.at(2), m_axis);
}

} // namespace internal
Expand Down

0 comments on commit 9bf85d3

Please sign in to comment.