gumbel_softmax_op.cc 3.9 KB