1917f71856
* Added support for validating custom layers. Fixes #928 * WIP #928 Added test for validating custom layers * WIP #928 Updated torch mock for custom layer