def __init__(self, node, initializer, inputs):
self.node = node
self.op_name = node.op_type
node_name = node.name.strip()
# it would be None if user doesn't set it manually
self.node_name = node_name if node_name else None
self.onnx_attr = OnnxHelper.parse_attr(
node.attribute) # dict(name: value)
self._initializer = initializer
self._input_list = inputs
self.model_inputs = self._extract_model_inputs()
self.model_trainable_values = self._extract_trainable_values()
self.output = node.output[0]
def __init__(self, node, _params, _all_tensors):
self.node = node
self.op_name = node.op_type
node_name = node.name.strip()
# it would be None if user doesn't set it manually
self.node_name = node_name if node_name else None
self.onnx_attr = OnnxHelper.parse_attr(
node.attribute) # dict(name: value)
self.params = [_params[i] for i in node.input if i in _params]
self.inputs = [
_all_tensors[i] for i in node.input if i in _all_tensors
]
self.all_tensors = _all_tensors
assert len(node.output) == 1, "we only support single output for now"
self.output = node.output[0]