Support more instructions in Hlo parser.
- while, tuple, send/recv, get-tuple-element, call. - "device=" Also, - Change HloModule::ToString to print computations in post order, so that a computation is defined before it's used. - Add % before computation name when it's used. PiperOrigin-RevId: 173350323
Loading
Please sign in to comment