output_uniform_quantized_tensor_type};
    auto new_conv_op_with_output_type =
        rewriter.create<stablehlo::ConvolutionOp>(
            op.getLoc(), new_conv_output_types, op.getOperands(),
            op->getAttrs());

    rewriter.replaceAllUsesWith(op.getResult(),
                                new_conv_op_with_output_type.getResult());

    auto new_output_dequant_op =

  // type
  auto new_device_cluster = builder.create<mlir::tf_device::ClusterOp>(
      device_cluster.getLoc(), device_result_types,
      /*operands=*/llvm::ArrayRef<Value>{}, device_cluster->getAttrs());

  new_device_cluster.getBody().push_back(new Block);
  builder.setInsertionPointToEnd(&new_device_cluster.GetBody());

  // Create return op for device computation region in the paralle_execute op