39 citations · 46 across the 2 of their papers we have counts for
3 papers
cs.PL2020★ 39 cited
Compiling ONNX Neural Network Models Using MLIR
Tian Jin, Gheorghe-Teodor Bercea, Tung D. Le +8
Deep neural network models are becoming increasingly popular and have been used in various tasks such as computer vision, speech recognition, and natural language processing. Machi…
cs.LG2018★ 7 cited
Fast and Accurate 3D Medical Image Segmentation with Data-swapping Method
Haruki Imai, Samuel Matzek, Tung D. Le +2
Deep neural network models used for medical image segmentation are large because they are trained with high-resolution three-dimensional (3D) images. Graphics processing units (GPU…
cs.LG2018
TFLMS: Large Model Support in TensorFlow by Graph Rewriting
Tung D. Le, Haruki Imai, Yasushi Negishi +1
While accelerators such as GPUs have limited memory, deep neural networks are becoming larger and will not fit with the memory limitation of accelerators for training. We propose a…