-
Notifications
You must be signed in to change notification settings - Fork 2.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add SegmentMax-16
to Core
#28698
base: master
Are you sure you want to change the base?
Add SegmentMax-16
to Core
#28698
Conversation
…into segmentmax_core Signed-off-by: p-wysocki <przemyslaw.wysocki@intel.com>
Signed-off-by: p-wysocki <przemyslaw.wysocki@intel.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The shape inference logic of the case when "num_segments" input is provided as Parameter needs adjustments (details in the comments).
src/core/shape_inference/include/segment_max_shape_inference.hpp
Outdated
Show resolved
Hide resolved
src/core/shape_inference/include/segment_max_shape_inference.hpp
Outdated
Show resolved
Hide resolved
src/core/shape_inference/include/segment_max_shape_inference.hpp
Outdated
Show resolved
Hide resolved
…into segmentmax_core Signed-off-by: p-wysocki <przemyslaw.wysocki@intel.com>
src/core/shape_inference/include/segment_max_shape_inference.hpp
Outdated
Show resolved
Hide resolved
src/plugins/intel_cpu/tests/unit/shape_inference_test/segment_max_shape_inference_test.cpp
Outdated
Show resolved
Hide resolved
src/plugins/intel_cpu/tests/unit/shape_inference_test/segment_max_shape_inference_test.cpp
Outdated
Show resolved
Hide resolved
src/plugins/intel_cpu/tests/unit/shape_inference_test/segment_max_shape_inference_test.cpp
Outdated
Show resolved
Hide resolved
Co-authored-by: Pawel Raasz <pawel.raasz@intel.com>
Co-authored-by: Pawel Raasz <pawel.raasz@intel.com>
Co-authored-by: Pawel Raasz <pawel.raasz@intel.com>
…max_shape_inference_test.cpp Co-authored-by: Pawel Raasz <pawel.raasz@intel.com>
Co-authored-by: Pawel Raasz <pawel.raasz@intel.com>
Co-authored-by: Pawel Raasz <pawel.raasz@intel.com>
Co-authored-by: Pawel Raasz <pawel.raasz@intel.com>
src/core/shape_inference/include/segment_max_shape_inference.hpp
Outdated
Show resolved
Hide resolved
src/core/shape_inference/include/segment_max_shape_inference.hpp
Outdated
Show resolved
Hide resolved
Co-authored-by: Katarzyna Mitrus <katarzyna.mitrus@intel.com>
… into segmentmax_core Signed-off-by: p-wysocki <przemyslaw.wysocki@intel.com>
std::vector<TRShape> shape_infer(const SegmentMax* op, | ||
const std::vector<TShape>& input_shapes, | ||
const ITensorAccessor& tensor_accessor = make_tensor_accessor()) { | ||
NODE_VALIDATION_CHECK(op, input_shapes.size() == 2 || input_shapes.size() == 3); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
NODE_VALIDATION_CHECK(op, input_shapes.size() == 2 || input_shapes.size() == 3); | |
OPENVINO_ASSERT(input_shapes.size() == 2 || input_shapes.size() == 3); |
Consider apply if there will be any other change in this PR
Details:
SegmentMax-16
to CoreRelated PRs:
SegmentMax-16
#28103SegmentMax-16
reference implementation #28788Tickets: