pytorch
549ef1d6 - [caffe][memonger] Extend operator schema check to dag memonger (#48021)

Commit
4 years ago
[caffe][memonger] Extend operator schema check to dag memonger (#48021) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/48021 Extending operator schema check for simple memonger to dag memonger as well. As part of this a fix is being made to handle inplace ops (having at least one output name same as input blob). Earlier all the output blobs from ops were being treated as shareable but it failed assertion of external input blobs with the same name not allowed to share. Test Plan: Added corresponding unit tests Reviewed By: hlu1 Differential Revision: D24968862 fbshipit-source-id: b6679a388a82b0d68f65ade64b85560354aaa3ef
Author
Parents
Loading