input / output
activation
reshape / cast
elementwise op
saved for backward (fp32)