Adding users of a node to the serialized JSON. (#59357)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/59357
Adding users of a node to the serialized JSON. Illustrated in the example:
JSON:
P419734894
Examples:
{
"shape": "[7]",
"dtype": "torch.float16",
"stride": "[1]",
"is_quantized": false,
"target": "conv.bias",
"op_code": "get_attr",
"name": "conv_bias",
"args": [],
"kwargs": {},
"users": [
{
"is_node": true,
"name": "to_dtype"
}
]
}
{
"target": "output",
"op_code": "output",
"name": "output",
"args": [
{
"is_node": true,
"name": "fba_layout_transform_1",
"shape": "[3, 7, 12, 12]",
"dtype": "torch.float16",
"stride": "[1008, 144, 12, 1]",
"is_quantized": false
}
],
"kwargs": {},
"users": []
}
Test Plan: buck test //caffe2/test:test_fx_experimental
Reviewed By: gcatron, jfix71
Differential Revision: D28857487
fbshipit-source-id: a3bac6bdb21ce10ba4a0d170c809aef13e6174a6
