llvm::enumerate(block.getArgumentTypes())) {
          TensorType updated_ty =
              UpdateElementTypeTo(original_ty, element_types[block_idx]);
          signature.addInputs(block_idx, {updated_ty});
        }
        rewriter.applySignatureConversion(&region.front(), signature);
      }
    }

    // Replace all uses of `op` results with the newly created op.