llvm::SmallVector<int64_t>{});
    mhlo::BroadcastInDimOp broadcast_in_dim_op =
        op_builder->create<mhlo::BroadcastInDimOp>(
            const_op->getLoc(), splat_elements_attr.getType(), scalar,
            broadcast_dims);
    const_op->replaceAllUsesWith(broadcast_in_dim_op);
    const_op->erase();
  }
};

}  // namespace