-
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
Add SegmentMax-16
to Core
#28698
Conversation
Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
…into segmentmax_core Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
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 <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
Co-authored-by: Pawel Raasz <[email protected]>
…max_shape_inference_test.cpp Co-authored-by: Pawel Raasz <[email protected]>
Co-authored-by: Pawel Raasz <[email protected]>
Co-authored-by: Pawel Raasz <[email protected]>
Co-authored-by: Pawel Raasz <[email protected]>
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 <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
… into segmentmax_core Signed-off-by: p-wysocki <[email protected]>
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: - Specification for `tf.math.segment_max` (https://www.tensorflow.org/api_docs/python/tf/math/segment_max) ### Related PRs: - #28698 - #28788 ### Tickets: - CVS-158914 --------- Signed-off-by: p-wysocki <[email protected]> Co-authored-by: Katarzyna Mitrus <[email protected]> Co-authored-by: Roman Kazantsev <[email protected]>
### Details: - Add reference implementation - Add tests ### Related PRs: - #28103 - #28698 - #28979 - #28999 ### Tickets: - CVS-158917 --------- Signed-off-by: p-wysocki <[email protected]> Co-authored-by: Roman Kazantsev <[email protected]> Co-authored-by: Pawel Raasz <[email protected]> Co-authored-by: Katarzyna Mitrus <[email protected]>
### Details: - The original PR (#28788) has been mistakenly force-merged due to a mistake in merge queue settings. It was later reverted, so this is the "new" Ref PR. - Add reference implementation - Add tests ### Related PRs: - #28103 - #28698 - #28979 - #28999 ### Tickets: - CVS-158917 --------- Signed-off-by: p-wysocki <[email protected]> Co-authored-by: Roman Kazantsev <[email protected]> Co-authored-by: Pawel Raasz <[email protected]> Co-authored-by: Katarzyna Mitrus <[email protected]>
Details:
SegmentMax-16
to CoreRelated PRs:
SegmentMax-16
#28103SegmentMax-16
reference implementation #28788SegmentMax-16
reference implementation #28979SegmentMax-16
#28999Tickets: