#include "tsl/lib/core/status_test_util.h"

namespace mlir::quant::stablehlo {
namespace {

using ::testing::Test;

class LegalizeTFQuantTest : public Test {
 protected:
  void TestBridgeLowering(llvm::StringRef mlir_module_string,
                          llvm::ArrayRef<tensorflow::TensorShape> arg_shapes,
                          tensorflow::DataType dtype) {
    tensorflow::tpu::MlirToHloArgs mlir_to_hlo_args;