inner_op.removeAttr(kAttrMapAttribute);
  }
  return success();
}

// Transfers the location of the main op in float function to ops with
// `attr_map` attributes in quantized function.
LogicalResult TransferLocation(func::FuncOp float_func,
                               func::FuncOp quantized_func) {
  Operation* main_op = nullptr;
  for (Operation& inner_op : float_func.getBody().front().getOperations()) {