Add a convenience method "forward_input_or_allocate_temp" to use buffer...
Add a convenience method "forward_input_or_allocate_temp" to use buffer forwarding for temporary tensors. Add buffer forwarding in a few additional places, including pooling ops and fused RNN kernels. Change: 149958948
Showing
想要评论请 注册 或 登录