namespace mlir {
namespace quant {

void RegisterOpsHook(TF::TensorFlowDialect &dialect) {
  dialect.addOperations<
#define GET_OP_LIST
#include "tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.cc.inc"
      >();
}

void RegisterOps() {
  TF_DIALECT_REGISTER_ADDITIONAL_OPERATIONS(RegisterOpsHook);
}

static auto kRegistration = (RegisterOps(), true);