xla
6270cbae - Add ShardingSpec to XLATensor when it is created with a PJRTShardedData (#5489)

Commit

2 years ago

Add ShardingSpec to XLATensor when it is created with a PJRTShardedData (#5489) * Add ShardingSpec to XLATensor when it is created with a PJRTShardedData * add test

References

#5489 - Add ShardingSpec to XLATensor when it is created with a PJRTShardedData

Author

JackCaoG

Parents

014caa53

xla 6270cbae - Add ShardingSpec to XLATensor when it is created with a PJRTShardedData (#5489)

xla
6270cbae - Add ShardingSpec to XLATensor when it is created with a PJRTShardedData (#5489)