experimental_experiment.torch_interpreter.oxs_opset

class experimental_experiment.torch_interpreter.oxs_opset.OxsOpset(builder: GraphBuilder)

Bridge with onnxscript.

Parameters:

builder – builder

make_node(op_type: str, *inputs: str | List[str] | None, outputs: int | List[str] | str | None = None, domain: str = '', name: str | None = None, **kwargs)

Creates a node.

Parameters:

• op_type – type

• inputs – inputs

• outputs – outputs

• domain – domain

• name – name

• kwargs – additional arguments

Returns:

output name

class experimental_experiment.torch_interpreter.oxs_opset.Var(name: str, builder: GraphBuilder | None = None)

Traceable variable name.