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.

Sign up for GitHub

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

Jump to bottom

Prevent division by 0 in OneHot implementation #49086

Merged
merged 1 commit into from May 11, 2021
Merged

Prevent division by 0 in OneHot implementation #49086

merged 1 commit into from May 11, 2021

Conversation

geetachavan1
Copy link
Contributor

If input indices is degenerate, the implementation would do a divide by zero. See

tensorflow/tensorflow/lite/kernels/one_hot.cc

Lines 68 to 72 in 745d57d

int prefix_dim_size = 1;
for (int i = 0; i < op_context.axis; ++i) {
prefix_dim_size *= op_context.indices->dims->data[i];
}
const int suffix_dim_size = NumElements(op_context.indices) / prefix_dim_size;

PiperOrigin-RevId: 370966870
Change-Id: Ie018337811c8016b5a1d3a277d00d5f2e19a2058

@mihaimaruseac @geetachavan1
Prevent division by 0 in OneHot implementation
4523630 
If input indices is degenerate, the implementation would do a divide by zero. See https://github.com/tensorflow/tensorflow/blob/745d57df6d5e9bc568666a2a48ed8dd629c27241/tensorflow/lite/kernels/one_hot.cc#L68-L72

PiperOrigin-RevId: 370966870
Change-Id: Ie018337811c8016b5a1d3a277d00d5f2e19a2058
@google-cla google-cla bot added the cla: yes label May 10, 2021
@mihaimaruseac mihaimaruseac merged commit b8679da into tensorflow:r2.5 May 11, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@mihaimaruseac mihaimaruseac mihaimaruseac approved these changes

Assignees
No one assigned
Labels
cla: yes
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
2 participants
@geetachavan1 @mihaimaruseac