PR #4904 Faster multinomial implementation

Faster multinomial implementation #4904

cowanmeg merged 2 commits into pytorch:master from cowanmeg:multinomial-v2

new multinomial implementation

9b873c23

JackCaoG requested a review from

Liyang90 2 years ago

JackCaoG requested a review from

wonjoo-wj 2 years ago

JackCaoG requested a review from

JackCaoG 2 years ago

wonjoo-wj approved these changes on 2023-04-19

wonjoo-wj commented on 2023-04-19

wonjoo-wj requested a review from

wonjoo-wj 2 years ago

fix for 2d input

1fb3bcc8

wonjoo-wj approved these changes on 2023-04-19

cowanmeg merged b1862ccc into master 2 years ago

miladm assigned

cowanmeg 2 years ago

miladm added lowering

miladm added performance

miladm requested a review from

miladm 2 years ago

Reviewers

wonjoo-wj

Liyang90

JackCaoG

miladm

Assignees

cowanmeg

Labels

lowering performance

Milestone

No milestone