Searched defs:y_draw (Results 1 – 1 of 1) sorted by relevance
1275 auto y_draw = F::gumbel_softmax(logits); in TEST_F() local1290 auto y_draw = F::gumbel_softmax( in TEST_F() local1305 auto y_draw = F::gumbel_softmax( in TEST_F() local1325 auto y_draw = F::gumbel_softmax( in TEST_F() local1345 torch::Tensor y_draw; in TEST_F() local