message: [XLA] Fix XLA profiler to consider expm1, log1p and atan2 as trigonometric function and display them under TFlops.
diff: @@ -102,7 +102,10 @@ Status HloCostAnalysis::HandleElementwiseOp(
  if (opcode == HloOpcode::kExp || opcode == HloOpcode::kLog ||
  opcode == HloOpcode::kPower || opcode == HloOpcode::kSqrt ||
  opcode == HloOpcode::kRsqrt || opcode == HloOpcode::kTanh ||
  - opcode == HloOpcode::kSin || opcode == HloOpcode::kCos) {
  + opcode == HloOpcode::kSin || opcode == HloOpcode::kCos ||
  + opcode == HloOpcode::kExpm1 || opcode == HloOpcode::kLog1p ||
  + opcode == HloOpcode::kAtan2) {
  +
  current_properties_[kTranscendentalsKey] = computation_count;
  } else {
  // Note: transcendental operations are considered a separate category from
  