Commit 903e7d93 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by Gunhan Gulsoy
Browse files

Improve the performance of ParseShapeStringInternal 

The previous implementation recompiled the shape regex at every call
what is an expensive opertaion. The new implementation improves the hlo
text parsing time for very large models for up to 9x by eliminating this
overhead.

PiperOrigin-RevId: 202454354
parent 38153493
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment