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 FusedEmbeddingSeqPoolGradKernel op for merge #19452

Closed
wants to merge 15 commits into from

Conversation

zhaify
Copy link
Contributor

@zhaify zhaify commented Aug 27, 2019

  • only support on Linux with Intel-MKL enabled
  • add padding_idx support
  • only support updating words' table with dense matrix
  • add UT for FW and BW both on with padding_idx or not

TODO

  • remove d_table->Resize(table_dim); in line 204

@zhaify
Copy link
Contributor Author

zhaify commented Aug 29, 2019

@yihuaxu please help to review this PR

@luotao1 luotao1 added the Intel label Aug 29, 2019
test=develop
@zhaify zhaify force-pushed the emb_seq_grad branch 7 times, most recently from d0d6cf3 to 0b3ea16 Compare September 11, 2019 01:07
test=develop
@luotao1
Copy link
Contributor

luotao1 commented Sep 17, 2019

close due to #19770

@luotao1 luotao1 closed this Sep 17, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants