xla
Add lowering for at::nan_to_num
#3093
Merged

Add lowering for at::nan_to_num #3093

JackCaoG merged 5 commits into pytorch:master from ymwangg:nan_to_num
ymwangg
JackCaoG JackCaoG requested a review from JackCaoG JackCaoG 4 years ago
ymwangg ymwangg force pushed from 0a4d3d45 to 234eeb88 4 years ago
JackCaoG
JackCaoG requested changes on 2021-08-20
ymwangg ymwangg force pushed from 271f25ce to 24f39823 4 years ago
ymwangg ymwangg force pushed from 24f39823 to b940b9ab 4 years ago
JackCaoG
JackCaoG requested changes on 2021-08-21
JackCaoG
JackCaoG requested changes on 2021-08-25
JackCaoG
ymwangg Add nan_to_num
349b659b
ymwangg Add unit test
a2d035ad
ymwangg Address PR comments
67ecd040
ymwangg Use ir::Value for replacement value
b4464abf
ymwangg Move dtype check earlier
48540363
ymwangg ymwangg force pushed from 030fcfed to 48540363 4 years ago
JackCaoG
JackCaoG approved these changes on 2021-08-26
JackCaoG JackCaoG merged bbbcbcd3 into master 4 years ago
ymwangg ymwangg deleted the nan_to_num branch 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone