Unverified Commit c80eb68a authored by Evgenya Stepyreva's avatar Evgenya Stepyreva Committed by GitHub

Dynamic Clamp shape/data_type propagation (#4282)

parent ef68ae31
...@@ -36,6 +36,7 @@ void op::Clamp::pre_validate_and_infer_types() ...@@ -36,6 +36,7 @@ void op::Clamp::pre_validate_and_infer_types()
{ {
NODE_VALIDATION_CHECK( NODE_VALIDATION_CHECK(
this, m_min < m_max, "The 'min' parameter needs to be less than 'max' for Clamp"); this, m_min < m_max, "The 'min' parameter needs to be less than 'max' for Clamp");
set_output_type(0, get_input_element_type(0), get_input_partial_shape(0));
} }
NodeVector op::Clamp::decompose_op() const NodeVector op::Clamp::decompose_op() const
......
...@@ -34,7 +34,7 @@ namespace ngraph ...@@ -34,7 +34,7 @@ namespace ngraph
class NGRAPH_API StridedSlice : public Op class NGRAPH_API StridedSlice : public Op
{ {
public: public:
static constexpr NodeTypeInfo type_info{"Slice", 1}; static constexpr NodeTypeInfo type_info{"StridedSlice", 1};
const NodeTypeInfo& get_type_info() const override { return type_info; } const NodeTypeInfo& get_type_info() const override { return type_info; }
StridedSlice() = default; StridedSlice() = default;
......
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