_format_snake_case_op_name(quantized_op))

    # Add suffix based on output type.
    suffix = '_fn'
    if len(arguments) > 1 and arguments[1] == 'f32':
      suffix = '_float_output_fn'
    function_name += suffix

    module = re.sub(compiled_regex, function_name, module, count=1)
  return module


def main(_: Sequence[str]) -> None:
  namespaces = _NAMESPACE.value.split('::')

constexpr StringRef kQuantizedOpsAttribute = "tf_quant.quantized_ops";
constexpr StringRef kCompositeFuncPrefix = "composite_";
constexpr StringRef kQuantizedFuncPrefix = "quantized_";
constexpr StringRef kFloatOutputFuncSuffix = "_float_output_fn";
constexpr StringRef kHybridFuncSuffix = "_hybrid_fn";

class QuantizeCompositeFunctionsPass
    : public mlir::PassWrapper<QuantizeCompositeFunctionsPass,
                               OperationPass<ModuleOp>> {