Function StepNumberForModule

tensorflow/compiler/xla/service/dump.cc:291–294

    *new absl::flat_hash_map<int64, uint64>();

int64 StepNumberForModule(const HloModule& module) {
  tensorflow::mutex_lock lock(mu);
  return module_id_to_step_number[module.unique_id()]++;
}
}  // namespace

string TimestampFor(const HloModule& module) {