基于OpenCL的TensorFlow框架中Element-Wise算子实现.docx

引言 深度学习模型以较强的建模性能逐渐被广泛应用于各类典型人工智能应用场景中。近年来,随着深度学习的发展,模型的网络结构大幅增加,复杂的神经网络结构增加了深度学习模型的拟合效果,同时使得模型的训练成本与日俱增。目前通过CPU提供的算力性价比较低且总体算力不足,因此目前通常利用异构并行设备满足深度学习模型的算力需求。目前主流深度学习框架,如TensorFlow[1]、PyTorch[2]、MXNet[3]等,均实现了对CUDA(Compute Unified Device Architecture)[4]和ROCm(Radeon Open Compute platform)[5]编程模型的支持,