pytorch
8ee1347c - Changes to support strides in addition to shape and dtype. (#56567)

Commit
3 years ago
Changes to support strides in addition to shape and dtype. (#56567) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56567 This adds stride information to the serialized JSON. This also adds shape, dtype and stride to the graph that is printed out. Test Plan: Run unit tests. Reviewed By: jfix71 Differential Revision: D27528988 fbshipit-source-id: f0be92055ad7c8e525625bfd1332c2db11ba612d
Author
Protonu Basu
Parents
Loading