gumbel_softmax_kernel_impl.h 3.5 KB