In this change we rename GetShardedInstructionSize to ByteSizeOfShapeIfShardedAcrossDevices.

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

Apache License 2.0

2.41k stars 361 forks source link

Closed copybara-service[bot] closed 1 week ago

copybara-service[bot] commented 1 week ago

In this change we rename GetShardedInstructionSize to ByteSizeOfShapeIfShardedAcrossDevices.