|
|
|
|
|
|
|
|
Argument: |
|
|
kind: union |
|
|
fields: |
|
|
as_none: |
|
|
type: bool |
|
|
as_tensor: |
|
|
type: TensorArgument |
|
|
as_tensors: |
|
|
type: List[TensorArgument] |
|
|
as_int: |
|
|
type: int |
|
|
as_ints: |
|
|
type: List[int] |
|
|
as_float: |
|
|
type: float |
|
|
as_floats: |
|
|
type: List[float] |
|
|
as_string: |
|
|
type: str |
|
|
as_strings: |
|
|
type: List[str] |
|
|
as_sym_int: |
|
|
type: SymIntArgument |
|
|
as_sym_ints: |
|
|
type: List[SymIntArgument] |
|
|
as_scalar_type: |
|
|
type: ScalarType |
|
|
as_memory_format: |
|
|
type: MemoryFormat |
|
|
as_layout: |
|
|
type: Layout |
|
|
as_device: |
|
|
type: Device |
|
|
as_bool: |
|
|
type: bool |
|
|
as_bools: |
|
|
type: List[bool] |
|
|
as_sym_bool: |
|
|
type: SymBoolArgument |
|
|
as_sym_bools: |
|
|
type: List[SymBoolArgument] |
|
|
as_graph: |
|
|
type: GraphArgument |
|
|
as_optional_tensors: |
|
|
type: List[OptionalTensorArgument] |
|
|
as_custom_obj: |
|
|
type: CustomObjArgument |
|
|
as_operator: |
|
|
type: str |
|
|
as_sym_float: |
|
|
type: SymFloatArgument |
|
|
as_sym_floats: |
|
|
type: List[SymFloatArgument] |
|
|
BufferMutationSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TensorArgument |
|
|
buffer_name: |
|
|
type: str |
|
|
ConstantValue: |
|
|
kind: union |
|
|
fields: |
|
|
as_none: |
|
|
type: bool |
|
|
as_int: |
|
|
type: int |
|
|
as_float: |
|
|
type: float |
|
|
as_string: |
|
|
type: str |
|
|
as_bool: |
|
|
type: bool |
|
|
CustomObjArgument: |
|
|
kind: struct |
|
|
fields: |
|
|
name: |
|
|
type: str |
|
|
class_fqn: |
|
|
type: str |
|
|
Device: |
|
|
kind: struct |
|
|
fields: |
|
|
type: |
|
|
type: str |
|
|
index: |
|
|
type: Optional[int] |
|
|
default: None |
|
|
ExportedProgram: |
|
|
kind: struct |
|
|
fields: |
|
|
graph_module: |
|
|
type: GraphModule |
|
|
opset_version: |
|
|
type: Dict[str, int] |
|
|
range_constraints: |
|
|
type: Dict[str, RangeConstraint] |
|
|
schema_version: |
|
|
type: SchemaVersion |
|
|
verifiers: |
|
|
type: List[str] |
|
|
default: '[]' |
|
|
torch_version: |
|
|
type: str |
|
|
default: <=2.4 |
|
|
GradientToParameterSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TensorArgument |
|
|
parameter_name: |
|
|
type: str |
|
|
GradientToUserInputSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TensorArgument |
|
|
user_input_name: |
|
|
type: str |
|
|
Graph: |
|
|
kind: struct |
|
|
fields: |
|
|
inputs: |
|
|
type: List[Argument] |
|
|
outputs: |
|
|
type: List[Argument] |
|
|
nodes: |
|
|
type: List[Node] |
|
|
tensor_values: |
|
|
type: Dict[str, TensorMeta] |
|
|
sym_int_values: |
|
|
type: Dict[str, SymInt] |
|
|
sym_bool_values: |
|
|
type: Dict[str, SymBool] |
|
|
is_single_tensor_return: |
|
|
type: bool |
|
|
default: 'False' |
|
|
custom_obj_values: |
|
|
type: Dict[str, CustomObjArgument] |
|
|
default: '{}' |
|
|
sym_float_values: |
|
|
type: Dict[str, SymFloat] |
|
|
default: '{}' |
|
|
GraphArgument: |
|
|
kind: struct |
|
|
fields: |
|
|
name: |
|
|
type: str |
|
|
graph: |
|
|
type: Graph |
|
|
GraphModule: |
|
|
kind: struct |
|
|
fields: |
|
|
graph: |
|
|
type: Graph |
|
|
signature: |
|
|
type: GraphSignature |
|
|
module_call_graph: |
|
|
type: List[ModuleCallEntry] |
|
|
metadata: |
|
|
type: Dict[str, str] |
|
|
default: '{}' |
|
|
GraphSignature: |
|
|
kind: struct |
|
|
fields: |
|
|
input_specs: |
|
|
type: List[InputSpec] |
|
|
output_specs: |
|
|
type: List[OutputSpec] |
|
|
InputSpec: |
|
|
kind: union |
|
|
fields: |
|
|
user_input: |
|
|
type: UserInputSpec |
|
|
parameter: |
|
|
type: InputToParameterSpec |
|
|
buffer: |
|
|
type: InputToBufferSpec |
|
|
tensor_constant: |
|
|
type: InputToTensorConstantSpec |
|
|
custom_obj: |
|
|
type: InputToCustomObjSpec |
|
|
token: |
|
|
type: InputTokenSpec |
|
|
constant_input: |
|
|
type: InputToConstantInputSpec |
|
|
InputToBufferSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TensorArgument |
|
|
buffer_name: |
|
|
type: str |
|
|
persistent: |
|
|
type: bool |
|
|
InputToConstantInputSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
name: |
|
|
type: str |
|
|
value: |
|
|
type: ConstantValue |
|
|
InputToCustomObjSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: CustomObjArgument |
|
|
custom_obj_name: |
|
|
type: str |
|
|
InputToParameterSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TensorArgument |
|
|
parameter_name: |
|
|
type: str |
|
|
InputToTensorConstantSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TensorArgument |
|
|
tensor_constant_name: |
|
|
type: str |
|
|
InputTokenSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TokenArgument |
|
|
Layout: |
|
|
kind: enum |
|
|
fields: |
|
|
Unknown: 0 |
|
|
SparseCoo: 1 |
|
|
SparseCsr: 2 |
|
|
SparseCsc: 3 |
|
|
SparseBsr: 4 |
|
|
SparseBsc: 5 |
|
|
_mkldnn: 6 |
|
|
Strided: 7 |
|
|
LossOutputSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TensorArgument |
|
|
MemoryFormat: |
|
|
kind: enum |
|
|
fields: |
|
|
Unknown: 0 |
|
|
ContiguousFormat: 1 |
|
|
ChannelsLast: 2 |
|
|
ChannelsLast3d: 3 |
|
|
PreserveFormat: 4 |
|
|
ModuleCallEntry: |
|
|
kind: struct |
|
|
fields: |
|
|
fqn: |
|
|
type: str |
|
|
signature: |
|
|
type: Optional[ModuleCallSignature] |
|
|
default: None |
|
|
ModuleCallSignature: |
|
|
kind: struct |
|
|
fields: |
|
|
inputs: |
|
|
type: List[Argument] |
|
|
outputs: |
|
|
type: List[Argument] |
|
|
in_spec: |
|
|
type: str |
|
|
out_spec: |
|
|
type: str |
|
|
forward_arg_names: |
|
|
type: Optional[List[str]] |
|
|
default: None |
|
|
NamedArgument: |
|
|
kind: struct |
|
|
fields: |
|
|
name: |
|
|
type: str |
|
|
arg: |
|
|
type: Argument |
|
|
Node: |
|
|
kind: struct |
|
|
fields: |
|
|
target: |
|
|
type: str |
|
|
inputs: |
|
|
type: List[NamedArgument] |
|
|
outputs: |
|
|
type: List[Argument] |
|
|
metadata: |
|
|
type: Dict[str, str] |
|
|
OptionalTensorArgument: |
|
|
kind: union |
|
|
fields: |
|
|
as_tensor: |
|
|
type: TensorArgument |
|
|
as_none: |
|
|
type: bool |
|
|
OutputSpec: |
|
|
kind: union |
|
|
fields: |
|
|
user_output: |
|
|
type: UserOutputSpec |
|
|
loss_output: |
|
|
type: LossOutputSpec |
|
|
buffer_mutation: |
|
|
type: BufferMutationSpec |
|
|
gradient_to_parameter: |
|
|
type: GradientToParameterSpec |
|
|
gradient_to_user_input: |
|
|
type: GradientToUserInputSpec |
|
|
user_input_mutation: |
|
|
type: UserInputMutationSpec |
|
|
token: |
|
|
type: OutputTokenSpec |
|
|
OutputTokenSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TokenArgument |
|
|
RangeConstraint: |
|
|
kind: struct |
|
|
fields: |
|
|
min_val: |
|
|
type: Optional[int] |
|
|
max_val: |
|
|
type: Optional[int] |
|
|
ScalarType: |
|
|
kind: enum |
|
|
fields: |
|
|
UNKNOWN: 0 |
|
|
BYTE: 1 |
|
|
CHAR: 2 |
|
|
SHORT: 3 |
|
|
INT: 4 |
|
|
LONG: 5 |
|
|
HALF: 6 |
|
|
FLOAT: 7 |
|
|
DOUBLE: 8 |
|
|
COMPLEXHALF: 9 |
|
|
COMPLEXFLOAT: 10 |
|
|
COMPLEXDOUBLE: 11 |
|
|
BOOL: 12 |
|
|
BFLOAT16: 13 |
|
|
UINT16: 28 |
|
|
SchemaVersion: |
|
|
kind: struct |
|
|
fields: |
|
|
major: |
|
|
type: int |
|
|
minor: |
|
|
type: int |
|
|
SymBool: |
|
|
kind: union |
|
|
fields: |
|
|
as_expr: |
|
|
type: SymExpr |
|
|
as_bool: |
|
|
type: bool |
|
|
SymBoolArgument: |
|
|
kind: union |
|
|
fields: |
|
|
as_name: |
|
|
type: str |
|
|
as_bool: |
|
|
type: bool |
|
|
SymExpr: |
|
|
kind: struct |
|
|
fields: |
|
|
expr_str: |
|
|
type: str |
|
|
hint: |
|
|
type: Optional[SymExprHint] |
|
|
default: None |
|
|
SymExprHint: |
|
|
kind: union |
|
|
fields: |
|
|
as_int: |
|
|
type: int |
|
|
as_bool: |
|
|
type: bool |
|
|
as_float: |
|
|
type: float |
|
|
SymFloat: |
|
|
kind: union |
|
|
fields: |
|
|
as_expr: |
|
|
type: SymExpr |
|
|
as_float: |
|
|
type: float |
|
|
SymFloatArgument: |
|
|
kind: union |
|
|
fields: |
|
|
as_name: |
|
|
type: str |
|
|
as_float: |
|
|
type: float |
|
|
SymInt: |
|
|
kind: union |
|
|
fields: |
|
|
as_expr: |
|
|
type: SymExpr |
|
|
as_int: |
|
|
type: int |
|
|
SymIntArgument: |
|
|
kind: union |
|
|
fields: |
|
|
as_name: |
|
|
type: str |
|
|
as_int: |
|
|
type: int |
|
|
TensorArgument: |
|
|
kind: struct |
|
|
fields: |
|
|
name: |
|
|
type: str |
|
|
TensorMeta: |
|
|
kind: struct |
|
|
fields: |
|
|
dtype: |
|
|
type: ScalarType |
|
|
sizes: |
|
|
type: List[SymInt] |
|
|
requires_grad: |
|
|
type: bool |
|
|
device: |
|
|
type: Device |
|
|
strides: |
|
|
type: List[SymInt] |
|
|
storage_offset: |
|
|
type: SymInt |
|
|
layout: |
|
|
type: Layout |
|
|
TokenArgument: |
|
|
kind: struct |
|
|
fields: |
|
|
name: |
|
|
type: str |
|
|
UserInputMutationSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: TensorArgument |
|
|
user_input_name: |
|
|
type: str |
|
|
UserInputSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: Argument |
|
|
UserOutputSpec: |
|
|
kind: struct |
|
|
fields: |
|
|
arg: |
|
|
type: Argument |
|
|
SCHEMA_VERSION: |
|
|
- 8 |
|
|
- 2 |
|
|
TREESPEC_VERSION: 1 |
|
|
|
