Unverified Commit fa0e7a4d authored by Scott Cyphers's avatar Scott Cyphers Committed by GitHub

Change attributes to int64_t (#4324)

Co-authored-by: 's avatarRobert Kimball <robert.kimball@intel.com>
Co-authored-by: 's avatarSang Ik Lee <sang.ik.lee@intel.com>
parent 0cf8b674
......@@ -35,8 +35,8 @@ op::v1::DeformableConvolution::DeformableConvolution(const Output<Node>& arg,
const CoordinateDiff& pads_end,
const Strides& dilations,
const PadType& auto_pad,
const size_t group,
const size_t deformable_group)
const int64_t group,
const int64_t deformable_group)
: Op({arg, deformable_values, filters})
, m_strides(strides)
, m_dilations(dilations)
......
......@@ -63,8 +63,8 @@ namespace ngraph
const CoordinateDiff& pads_end,
const Strides& dilations,
const PadType& auto_pad = PadType::EXPLICIT,
const size_t group = 1,
const size_t deformable_group = 1);
const int64_t group = 1,
const int64_t deformable_group = 1);
bool visit_attributes(AttributeVisitor& visitor) override;
void validate_and_infer_types() override;
......@@ -79,10 +79,10 @@ namespace ngraph
void set_pads_end(const CoordinateDiff& pads_end) { m_pads_end = pads_end; }
const PadType& get_auto_pad() const { return m_auto_pad; }
void set_auto_pad(const PadType& auto_pad) { m_auto_pad = auto_pad; }
size_t get_group() const { return m_group; }
void set_group(const size_t group) { m_group = group; }
size_t get_deformable_group() const { return m_deformable_group; }
void set_deformable_group(const size_t deformable_group)
int64_t get_group() const { return m_group; }
void set_group(const int64_t group) { m_group = group; }
int64_t get_deformable_group() const { return m_deformable_group; }
void set_deformable_group(const int64_t deformable_group)
{
m_deformable_group = deformable_group;
}
......@@ -96,8 +96,8 @@ namespace ngraph
CoordinateDiff m_pads_begin;
CoordinateDiff m_pads_end;
PadType m_auto_pad;
size_t m_group;
size_t m_deformable_group;
int64_t m_group;
int64_t m_deformable_group;
};
}
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment