def args_check(cls, node, **kwargs): if node.op_type == "SplitV": if node.inputs[2] not in kwargs["consts"]: exception.CONST_NOT_FOUND_EXCEPT(node.inputs[2], node.op_type) if node.op_type == "Split": if node.inputs[0] not in kwargs["consts"]: exception.CONST_NOT_FOUND_EXCEPT(node.inputs[0], node.op_type)
def args_check(cls, node, **kwargs): if node.inputs[1] not in kwargs["consts"]: exception.CONST_NOT_FOUND_EXCEPT(node.inputs[1], node.op_type) supported_modes = ["constant", "reflect"] mode = node.attr.get("mode", "constant") if mode.lower() not in supported_modes: raise RuntimeError( "Mode {} of Pad is not supported in ONNX.".format(mode))
def args_check(cls, node, **kwargs): if node.inputs[3] not in kwargs["consts"]: exception.CONST_NOT_FOUND_EXCEPT(node.inputs[3], node.op_type)
def args_check(cls, node, **kwargs): if cls.SINCE_VERSION == 2: if node.inputs[1] not in kwargs["consts"]: exception.CONST_NOT_FOUND_EXCEPT(node.inputs[1], node.op_type) if node.inputs[2] not in kwargs["consts"]: exception.CONST_NOT_FOUND_EXCEPT(node.inputs[2], node.op_type)