"mlir::stablehlo::StablehloDialect",
      "TF::TensorFlowDialect",
      "mlir::quant::QuantizationDialect",
      "mlir::quantfork::QuantizationForkDialect",
  ];
}

def FoldConstantTransposePass : Pass<"stablehlo-fold-constant-transpose", "mlir::func::FuncOp"> {
  let summary = "Folds stablehlo.constant -> stablehlo.transpose patterns.";
  let description = [{

// to match the function signature. Constants are also transpose-folded
// accordingly.
//
// Note that the `transpose` after the `broadcast_in_dim` is not folded by the
// `FoldConstantTransposePass`.

// CHECK-LABEL: nchw_conv_with_broadcasted_bias_add_relu
// CHECK-SAME: %[[ARG:.+]]: tensor<1x2x5x5xf32>
func.func @nchw_conv_with_broadcasted_bias_add_relu(%arg0: tensor<1x2x5x5xf32>) -> tensor<1x4x5x5xf32> {