gumbel_softmax_kernel_impl.h 3.3 KB