Skip to content
This repository was archived by the owner on Jan 3, 2023. It is now read-only.

Commit 2aa508b

Browse files
Mateusz Bencerdiyessi
authored andcommitted
Adjust op ROIPooling (#3346)
* Adjust ROIPolling * Fixed comment
1 parent 0850d23 commit 2aa508b

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

src/ngraph/op/experimental/layers/roi_pooling.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@ using namespace ngraph;
2222
const string op::ROIPooling::type_name{"ROIPooling"};
2323

2424
op::ROIPooling::ROIPooling(const shared_ptr<Node>& input,
25-
const std::shared_ptr<Node>& coords,
25+
const shared_ptr<Node>& coords,
2626
const Shape& output_size,
2727
const float spatial_scale,
28-
const std::string& kind)
28+
const string& method)
2929
: Op(check_single_output_args({input, coords}))
3030
, m_output_size(output_size)
3131
, m_spatial_scale(spatial_scale)
32-
, m_kind(kind)
32+
, m_method(method)
3333
{
3434
constructor_validate_and_infer_types();
3535
}
@@ -69,5 +69,5 @@ shared_ptr<Node> op::ROIPooling::copy_with_new_args(const NodeVector& new_args)
6969
{
7070
check_new_args_count(this, new_args);
7171
return make_shared<ROIPooling>(
72-
new_args.at(0), new_args.at(1), m_output_size, m_spatial_scale, m_kind);
72+
new_args.at(0), new_args.at(1), m_output_size, m_spatial_scale, m_method);
7373
}

src/ngraph/op/experimental/layers/roi_pooling.hpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,12 +34,12 @@ namespace ngraph
3434
/// \param coords Coordinates of bounding boxes
3535
/// \param output_size Height/Width of ROI output features
3636
/// \param spatial_scale Ratio of input feature map over input image size
37-
/// \param kind Kind of pooling - Max or Bilinear
37+
/// \param method Method of pooling - Max or Bilinear
3838
ROIPooling(const std::shared_ptr<Node>& input,
3939
const std::shared_ptr<Node>& coords,
4040
const Shape& output_size,
4141
const float spatial_scale,
42-
const std::string& kind);
42+
const std::string& method);
4343

4444
void validate_and_infer_types() override;
4545

@@ -48,11 +48,11 @@ namespace ngraph
4848

4949
const Shape& get_output_size() const { return m_output_size; }
5050
float get_spatial_scale() const { return m_spatial_scale; }
51-
const std::string& get_kind() const { return m_kind; }
51+
const std::string& get_method() const { return m_method; }
5252
private:
5353
Shape m_output_size;
5454
float m_spatial_scale;
55-
std::string m_kind;
55+
std::string m_method;
5656
};
5757
}
5858
}

0 commit comments

Comments
 (0)