xla
d1748e0f - [LTC] Adopt even more LazyTensor interfaces (#4317)

Commit
2 years ago
[LTC] Adopt even more LazyTensor interfaces (#4317) Summary: This patch adopts even more LazyTensor interfaces: 1. Reuse TryLimitGraphSize and CreateTensorNode directly. 2. Override size, ToTensor, dtype, ApplyPendingGraph, AssignIrValue, and GetIrValueForTensor. It depends on pytorch/pytorch#90686. Test Plan: CI.
Author
Parents
Loading