Sean Moriarity
Sean Moriarity
Right now we just raise the Nx error, but we should do a better job of making it clear why the mismatch error occurs and possible options to fix it
I think this would enable some pretty interesting functionality, but I haven't yet thought through any potential drawbacks. I will do a detailed proposal soon
There is no need to include these because they are not a part of the optimization process, so it ends up being a waste of memory duplicating them and for...
Checklist for serialization support of Axon models. ## Linear - [x] `dense` - [ ] `bilinear` - [ ] `embedding` ## Convolutional - [x] `conv` - [ ] `conv_transpose` -...
ONNX pooling operations require that the padding configuration have less padding on each side than the size of the kernel. This isn't validated currently but should be.