message: Call HloComputation.Accept instead of HloInstruction.Accept to get all instructions profiled.
  RELNOTES: n/a
diff: @@ -198,8 +198,8 @@ class CollectProfileCandidates : public DfsHloVisitorWithDefault {
  std::unordered_map<const HloInstruction*, size_t> hlo_to_profile_idx;
  CollectProfileCandidates profile_candidates_for_computation(
  &hlo_to_profile_idx);
  - TF_RETURN_IF_ERROR(computation->root_instruction()->Accept(
  - &profile_candidates_for_computation));
  + TF_RETURN_IF_ERROR(
  + computation->Accept(&profile_candidates_for_computation));
  return hlo_to_profile_idx;
  }
  