ComposeUniformQuantizedTypePass> {
 private:
  void runOnOperation() override;
};

// Tests whether a `stablehlo::ConvertOp` is a i8 -> f32 cast.
bool IsI8ToF32Cast(stablehlo::ConvertOp convert_op) {
  const bool is_i8_operand =
      convert_op.getOperand().getType().getElementType().isInteger(/*width=*/8);
  const bool is_f32_result =