onnx pytorch QAT converter
代码:
def construct_producer_consumers(self):
for node in self._onnx_model.graph.node:
for input_tensor in node.input:
if input_tensor not in self._consumers:
self._consumers[input_tensor] = []
self._consumers[input_tensor].append(node)
for output_tensor in node.output:
self._producer[output_tensor] = node
_consumers: key是上游结点,value是下游结点的list。因为一个上游可能 作为多个下游结点的输入,所以valu是一个list
_producer:key是下游结点,value是上游结点。因为一个下游结点只可能被一个人产生,不可能有多个结点生产出同样的output结点,所以value是一个结点,不是list。
好