onnxruntime
Assert that the data is on the same device as ORTModule
#6942
Merged

Assert that the data is on the same device as ORTModule #6942

baijumeswani
baijumeswani Assert that the data and module devices are the same
a07baef6
baijumeswani baijumeswani requested a review from BowenBao BowenBao 5 years ago
baijumeswani baijumeswani requested a review from liqunfu liqunfu 5 years ago
baijumeswani baijumeswani requested a review from spandantiwari spandantiwari 5 years ago
baijumeswani baijumeswani requested a review from thiagocrepaldi thiagocrepaldi 5 years ago
thiagocrepaldi
thiagocrepaldi approved these changes on 2021-03-08
baijumeswani Check if input is a tensor before checking device
a5706e08
baijumeswani Change _check_device to _check_same_device
56aadcb1
thiagocrepaldi
thiagocrepaldi approved these changes on 2021-03-09
SherlockNoMad
SherlockNoMad SherlockNoMad added component:ortmodule
baijumeswani baijumeswani merged f1ade14e into thiagofc/ortmodule-api 5 years ago
baijumeswani baijumeswani deleted the bmeswani/data_device_check branch 5 years ago
baijumeswani

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone