* Fixes for build with the new pytorch * Fixes for build with the new pytorch * Fixes for build with the new pytorch
