Skip to content

Add support for PReLU of 2d tensors

Sam Eadie requested to merge sameadie/sadl:prelu2d into master

Currently, the non-SIMD implementation of PReLU expects inputs to be 4d tensors and fails silently for other shapes such as 2d tensors since it's init function does not check the input's size. This MR fixes this init and provides a non-SIMD implementation for 2d tensors.

Merge request reports