pytorch
e454be5f - [MPS][BE] Introduce MPSUnaryCachedGraph (#81033)

Commit
3 years ago
[MPS][BE] Introduce MPSUnaryCachedGraph (#81033) I.e. CachedGraph that has input and output tensors Also, add `MPSGraphCache::LookUpAs` template, which combines LookUp with static_cast to target type Pull Request resolved: https://github.com/pytorch/pytorch/pull/81033 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading