Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

OP request: GPU version of mxnet.ndarray.contrib.boolean_mask #13675

Closed
kalpitdixit opened this issue Dec 18, 2018 · 2 comments
Closed

OP request: GPU version of mxnet.ndarray.contrib.boolean_mask #13675

kalpitdixit opened this issue Dec 18, 2018 · 2 comments
Labels
Feature request Gluon

Comments

@kalpitdixit
Copy link

kalpitdixit commented Dec 18, 2018 •
edited
Loading

I need to use the "boolean_mask" operator in my Gluon model. Currently, it is only implemented for CPU which is making training for my model very slow (multiple weeks). Can we please have this op in GPU as well?

https://mxnet.incubator.apache.org/api/python/ndarray/contrib.html#mxnet.ndarray.contrib.boolean_mask

@szha
@andrewfayres
Copy link
Contributor

@mxnet-label-bot add [gluon, feature request]

@szha
Copy link
Member

szha commented Feb 19, 2019

#14090

@szha szha closed this as completed Feb 19, 2019
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
Feature request Gluon
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
4 participants
@szha @andrewfayres @kalpitdixit @marcoabreu