PR #10164: [XLA:CPU] Improve Trace emitter
PR #10164: [XLA:CPU] Improve Trace emitter

Imported from GitHub PR https://github.com/openxla/xla/pull/10164

This PR improves tracing a bit.

Without this PR, you would see operations in trace view as :
![image](https://github.com/openxla/xla/assets/43043975/f5ae19c9-851d-44fe-bf75-9a456425a387)

This way , it s difficult to see a) target of custom call b) can't group operations based on type of hlo operation and hence makes op list huge

This PR addresses these problems and now you should see tensorboard trace view as : 
![image](https://github.com/openxla/xla/assets/43043975/5a10b3ef-ea61-4ac5-90e3-92ead1354aa4)

Copybara import of the project:

--
8cd6363890107ce7af2ed3630d577ad4d20a745b by Sachin Muradi <sachin.muradi@intel.com>:

Update Trace emitter

--
a2813ad7439da8934b98b12623193c3b120eb9ca by Sachin Muradi <sachin.muradi@intel.com>:

more update

Merging this change closes #10164

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10164 from Intel-tensorflow:sachin/tracemeupdate a2813ad7439da8934b98b12623193c3b120eb9ca
