Commit 5f173dbf authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Make the HloDomainRemover pass more configurable

Previously we had two different function to normalize instructions
within a domain where one of them was specified inside the metadata
while the other one is passed into the domain remover.

This change unifies them to use the externally passed in function for
both usecase to make it possible to rewrite both of them from the caller
of the domain remover (to add special logic).

PiperOrigin-RevId: 204715075
parent eadcdf91
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment