Conv and Pool layers
Benny-Nottonson opened this issue · comments
Benny Nottonson commented
Is your feature request related to a problem? Please describe.
Layers and functions that rely on a kernel and a main tensor are currently not supported, these need to be added for Conv and Pool to work.
Describe the solution you'd like
These should be similar in structure to operation functions since they take in 3 nodes and store data in the first, but more research needs to be done to optimize these, potentially using tiling or vectorized operations in parallel.