Du Li 621b3ac03a FFT contrib ops (#3381) ... * add custom op skeleton * Adding Rfft, Irfft kernels. * Fix a few errors: 1. make kernel stateless to avoid race condition 2. reclaim cufft plan * Adding MLFloat16 support * Adding fp16 support for fft ops. * Adding cufft plan cache. * adding a util func * adding copyright info. * Accommodating PR comments.		2020-04-14 10:12:04 -07:00
..
cpu	Fix crash in DequantizeLinear with scalar tensor (#3508)	2020-04-13 14:52:52 -07:00
cuda	FFT contrib ops (#3381)	2020-04-14 10:12:04 -07:00
cpu_contrib_kernels.cc	Reduce binary size of Slice implementation (#3238)	2020-04-08 07:19:29 +10:00
cpu_contrib_kernels.h	Fix kernel registry bug (#2137)	2019-10-17 23:10:54 -07:00
cuda_contrib_kernels.cc	FFT contrib ops (#3381)	2020-04-14 10:12:04 -07:00
cuda_contrib_kernels.h	move all contrib ops to contrib ops namespace (#1190)	2019-06-24 10:19:01 -07:00