xla
d7b80a4e - Add ShardingSpec to XLATensor when it is created with a PJRTShardedData

Commit
2 years ago
Add ShardingSpec to XLATensor when it is created with a PJRTShardedData
Author
Parents
Loading