I can find the Linear routines to run the forward calculations and calculate gradients, e.g.
grep -r ‘(Linear’ pytorch/aten | grep API
pytorch/aten/src/THNN/generic/THNN.h:TH_API void THNN_(Linear_updateOutput)(
pytorch/aten/src/THNN/generic/THNN.h:TH_API void THNN_(Linear_updateGradInput)(
pytorch/aten/src/THNN/generic/THNN.h:TH_API void THNN_(Linear_accGradParameters)(
but can’t locate the equivalent routines in THCUNN
?
thanks