3 citations · 3 across the 1 of their papers we have counts for
3 papers
cs.CV2022★ 3 cited
GPUNet: Searching the Deployable Convolution Neural Networks for GPUs
Linnan Wang, Chenhan Yu, Satish Salian +3
Customizing Convolution Neural Networks (CNN) for production use has been a challenging task for DL practitioners. This paper intends to expedite the model customization with a mod…
math.NA2019
Fast Approximation of the Gauss-Newton Hessian Matrix for the Multilayer Perceptron
Chao Chen, Severin Reiz, Chenhan Yu +2
We introduce a fast algorithm for entry-wise evaluation of the Gauss-Newton Hessian (GNH) matrix for the fully-connected feed-forward neural network. The algorithm has a precomputa…
cs.MS2018
Implementing Strassen's Algorithm with CUTLASS on NVIDIA Volta GPUs
Jianyu Huang, Chenhan D. Yu, Robert A. van de Geijn
Conventional GPU implementations of Strassen's algorithm (Strassen) typically rely on the existing high-performance matrix multiplication (GEMM), trading space for time. As a resul…