return False
if child.op_type != "Add":
return False
if "broadcast" not in child.attrs:
return False
if child.attrs["broadcast"] != 1:
return False
if "axis" not in child.attrs:
return False
After Change
if child.inputs[1] not in child.input_tensors:
return False
t = child.input_tensors[child.inputs[1]]
if len(np.squeeze(t).shape) != 1:
return False
if parent.inputs[1] not in parent.input_tensors:
return False