MACE代码总结

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。

1 thought on “MACE代码总结”

Leave a Comment