google-research-datasets / tpu_graphs

Apache License 2.0
120 stars 43 forks source link

Generating raw graph data in Protobuf format #12

Open rahulvaidya20 opened 2 months ago

rahulvaidya20 commented 2 months ago

Hi @mangpo and @samihaija ,

Given a tensorflow program, I'm trying to generate the raw graph data (hoo) in protobuf and extract its features. I'm only looking for the default configuration (without autotuner/search space exploration). I tried generating HLO graph using tf’s experimental_get_compiler_ir function, but I don't think this graph format represents the pb format, defined in tuning.proto. It would be great if you can help me in this regard.