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