12 Commits (8c19d7aa2f89a38b3a68e53c73d88af16a3de8ce)

Author	SHA1	Message	Date
jakpiase	5b4f8aac82	Added LSTM BF16 and fixed GRU BF16 (#31234)	4 years ago
jakpiase	f8da5536ed	REUPLOAD Added vanilla LSTM and LSTM with peepholes oneDNN fp32 kernel (#30719) ... * added external reorder to profiler * resolved conflict * added enable_static * initial version of lstm, not working yet * added lstm to operators.cmake * added vanilla lstm mkldnn op * added peephole weights integration * minor changes * added formatting * added fusion_lstm_mkldnn to static_whitelist * added formatting * removed comment * moved use_peepholes attribute inside is_cached block * reverted wrong changes * minor formatting change * minor changes * changed stream handling * minor change * added datatype to GetExpectedKernelType() * added reading stream from TLS	4 years ago
Tao Luo	824a79d383	Revert "Added vanilla LSTM and LSTM with peepholes oneDNN fp32 kernel (#30661)" (#30708) ... This reverts commit d834f4e6e8.	4 years ago
jakpiase	d834f4e6e8	Added vanilla LSTM and LSTM with peepholes oneDNN fp32 kernel (#30661) ... * added external reorder to profiler * resolved conflict * added enable_static * initial version of lstm, not working yet * added lstm to operators.cmake * added vanilla lstm mkldnn op * added peephole weights integration * minor changes * added formatting * added fusion_lstm_mkldnn to static_whitelist * added formatting * removed comment * moved use_peepholes attribute inside is_cached block * reverted wrong changes * minor formatting change * minor changes	4 years ago
Jacek Czaja	173660be7b	[oneDNN] Cache oneDNN stream not to recreate in each oneDNN op (#30358)	4 years ago
Jacek Czaja	f6cca62575	[oneDNN] Making ThreadID info in caching key optional (#29272)	4 years ago
Wojciech Uss	04bcc13fac	Add multi_gru op and tests (#28591) ... * Add multi_gru op and tests * removed redundant disable_dygraph()	4 years ago
Jacek Czaja	606611d351	[oneDNN] GRU BF16 kernel (#27731)	4 years ago
Wojciech Uss	966447e338	Added support for quantization of fusion_gru (#27518)	4 years ago
Adam	cc3f4b813a	Add int8 GRU kernel (#27220) ... * Add int8 GRU kernel with UTs * Lint fixes * More lint fixes	4 years ago
lidanqing	5c4eed66fd	Fix GRU mkldnn kernel fail on look_table_v2 (#27198) ... * Fix the lookup_table_v2 failed on GRU mkldnn kernel issue test=develop * fix according to reviews, removed x_num_col_dims test=develop * update gru model. change according to reviews test=develop * change according to reviews test=develop	4 years ago
Adam	68c6160e63	Add oneDNN fusion_gru kernel (#25594) ... * Add oneDNN fusion_gru kernel and fix fc+gru pass test=develop * Formatting changes test=develop * Lint fixes test=develop * Add memory::format_tag::any to GRU weights test=develop * Fix build with CUDA * Fix build with CUDA v2	5 years ago