RobinGrosman 5d5225f2ee optimize fasterrcnn on gpu. fix randomchoicewithmask, topk kernels		2021-02-09 13:57:00 -08:00
..
random_categorical_gpu_kernel.cc	add dtype supports to random_categorical on gpu	2020-12-07 15:11:25 +08:00
random_categorical_gpu_kernel.h	!9519 Add dtype supports to op random_categorical on gpu	2020-12-07 21:51:03 +08:00
random_choice_with_mask_gpu_kernel.cc	add GPU support to RandomChoiceWithMask	2020-08-01 00:33:59 -04:00
random_choice_with_mask_gpu_kernel.h	optimize fasterrcnn on gpu. fix randomchoicewithmask, topk kernels	2021-02-09 13:57:00 -08:00
uniform_candidate_sampler_gpu_kernel.cc	add int64 support to UniformCandidateSampler GPU	2020-12-10 11:08:33 -05:00
uniform_candidate_sampler_gpu_kernel.h	SampledSoftmaxLoss stuck issue	2020-12-17 08:53:03 -05:00