Skip to content
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

Merged
merged 26 commits into from
Feb 13, 2025
Merged

Conversation

Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
@p-wysocki p-wysocki added this to the 2025.1 milestone Jan 27, 2025
@p-wysocki p-wysocki requested review from a team as code owners January 27, 2025 15:33
@github-actions github-actions bot added category: Core OpenVINO Core (aka ngraph) category: CPU OpenVINO CPU plugin category: CPP API OpenVINO CPP API bindings labels Jan 27, 2025
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]>
Copy link
Contributor

@mitruska mitruska left a 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).

Signed-off-by: p-wysocki <[email protected]>
p-wysocki and others added 3 commits February 6, 2025 00:18
Co-authored-by: Katarzyna Mitrus <[email protected]>
Signed-off-by: p-wysocki <[email protected]>
@p-wysocki p-wysocki requested a review from praasz February 5, 2025 23:53
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);
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
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

@p-wysocki p-wysocki added this pull request to the merge queue Feb 12, 2025
github-merge-queue bot pushed a commit that referenced this pull request Feb 12, 2025
### 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]>
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Feb 12, 2025
@p-wysocki p-wysocki added this pull request to the merge queue Feb 12, 2025
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Feb 12, 2025
@p-wysocki p-wysocki enabled auto-merge February 12, 2025 14:39
@p-wysocki p-wysocki added this pull request to the merge queue Feb 12, 2025
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Feb 12, 2025
@mlukasze mlukasze added this pull request to the merge queue Feb 13, 2025
Merged via the queue into openvinotoolkit:master with commit d455dd2 Feb 13, 2025
185 checks passed
github-merge-queue bot pushed a commit that referenced this pull request Feb 17, 2025
### 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]>
github-merge-queue bot pushed a commit that referenced this pull request Feb 18, 2025
### 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]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
category: Core OpenVINO Core (aka ngraph) category: CPP API OpenVINO CPP API bindings category: CPU OpenVINO CPU plugin
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants