• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# @generated by update_schema.py
2# checksum<<923abf371a1f8802cacb037d409d28273867777a98f6542fba28616c2b92b639>>
3Argument:
4  kind: union
5  fields:
6    as_none:
7      type: Tuple[()]
8    as_tensor:
9      type: TensorArgument
10    as_tensors:
11      type: List[TensorArgument]
12    as_int:
13      type: int
14    as_ints:
15      type: List[int]
16    as_float:
17      type: float
18    as_floats:
19      type: List[float]
20    as_string:
21      type: str
22    as_strings:
23      type: List[str]
24    as_sym_int:
25      type: SymIntArgument
26    as_sym_ints:
27      type: List[SymIntArgument]
28    as_scalar_type:
29      type: ScalarType
30    as_memory_format:
31      type: MemoryFormat
32    as_layout:
33      type: Layout
34    as_device:
35      type: Device
36    as_bool:
37      type: bool
38    as_bools:
39      type: List[bool]
40    as_sym_bool:
41      type: SymBoolArgument
42    as_sym_bools:
43      type: List[SymBoolArgument]
44    as_graph:
45      type: GraphArgument
46    as_optional_tensors:
47      type: List[OptionalTensorArgument]
48    as_custom_obj:
49      type: CustomObjArgument
50    as_operator:
51      type: str
52BufferMutationSpec:
53  kind: struct
54  fields:
55    arg:
56      type: TensorArgument
57    buffer_name:
58      type: str
59ConstantInputSpec:
60  kind: struct
61  fields:
62    name:
63      type: str
64    value:
65      type: ConstantValue
66ConstantValue:
67  kind: union
68  fields:
69    as_none:
70      type: Tuple[()]
71    as_int:
72      type: int
73    as_float:
74      type: float
75    as_string:
76      type: str
77    as_bool:
78      type: bool
79CustomObjArgument:
80  kind: struct
81  fields:
82    name:
83      type: str
84    class_fqn:
85      type: str
86Device:
87  kind: struct
88  fields:
89    type:
90      type: str
91    index:
92      type: Optional[int]
93      default: None
94ExportedProgram:
95  kind: struct
96  fields:
97    graph_module:
98      type: GraphModule
99    opset_version:
100      type: Dict[str, int]
101    range_constraints:
102      type: Dict[str, RangeConstraint]
103    schema_version:
104      type: SchemaVersion
105    verifiers:
106      type: List[str]
107      default: '[]'
108    torch_version:
109      type: str
110      default: <=2.4
111GradientToParameterSpec:
112  kind: struct
113  fields:
114    arg:
115      type: TensorArgument
116    parameter_name:
117      type: str
118GradientToUserInputSpec:
119  kind: struct
120  fields:
121    arg:
122      type: TensorArgument
123    user_input_name:
124      type: str
125Graph:
126  kind: struct
127  fields:
128    inputs:
129      type: List[Argument]
130    outputs:
131      type: List[Argument]
132    nodes:
133      type: List[Node]
134    tensor_values:
135      type: Dict[str, TensorMeta]
136    sym_int_values:
137      type: Dict[str, SymInt]
138    sym_bool_values:
139      type: Dict[str, SymBool]
140    is_single_tensor_return:
141      type: bool
142      default: 'False'
143    custom_obj_values:
144      type: Dict[str, CustomObjArgument]
145      default: '{}'
146GraphArgument:
147  kind: struct
148  fields:
149    name:
150      type: str
151    graph:
152      type: Graph
153GraphModule:
154  kind: struct
155  fields:
156    graph:
157      type: Graph
158    signature:
159      type: GraphSignature
160    module_call_graph:
161      type: List[ModuleCallEntry]
162    metadata:
163      type: Dict[str, str]
164      default: '{}'
165GraphSignature:
166  kind: struct
167  fields:
168    input_specs:
169      type: List[InputSpec]
170    output_specs:
171      type: List[OutputSpec]
172InputSpec:
173  kind: union
174  fields:
175    user_input:
176      type: UserInputSpec
177    parameter:
178      type: InputToParameterSpec
179    buffer:
180      type: InputToBufferSpec
181    tensor_constant:
182      type: InputToTensorConstantSpec
183    custom_obj:
184      type: InputToCustomObjSpec
185    token:
186      type: InputTokenSpec
187    constant_input:
188      type: ConstantInputSpec
189InputToBufferSpec:
190  kind: struct
191  fields:
192    arg:
193      type: TensorArgument
194    buffer_name:
195      type: str
196    persistent:
197      type: bool
198InputToCustomObjSpec:
199  kind: struct
200  fields:
201    arg:
202      type: CustomObjArgument
203    custom_obj_name:
204      type: str
205InputToParameterSpec:
206  kind: struct
207  fields:
208    arg:
209      type: TensorArgument
210    parameter_name:
211      type: str
212InputToTensorConstantSpec:
213  kind: struct
214  fields:
215    arg:
216      type: TensorArgument
217    tensor_constant_name:
218      type: str
219InputTokenSpec:
220  kind: struct
221  fields:
222    arg:
223      type: TokenArgument
224Layout:
225  kind: enum
226  fields:
227    Unknown: 0
228    SparseCoo: 1
229    SparseCsr: 2
230    SparseCsc: 3
231    SparseBsr: 4
232    SparseBsc: 5
233    _mkldnn: 6
234    Strided: 7
235LossOutputSpec:
236  kind: struct
237  fields:
238    arg:
239      type: TensorArgument
240MemoryFormat:
241  kind: enum
242  fields:
243    Unknown: 0
244    ContiguousFormat: 1
245    ChannelsLast: 2
246    ChannelsLast3d: 3
247    PreserveFormat: 4
248ModuleCallEntry:
249  kind: struct
250  fields:
251    fqn:
252      type: str
253    signature:
254      type: Optional[ModuleCallSignature]
255      default: None
256ModuleCallSignature:
257  kind: struct
258  fields:
259    inputs:
260      type: List[Argument]
261    outputs:
262      type: List[Argument]
263    in_spec:
264      type: str
265    out_spec:
266      type: str
267NamedArgument:
268  kind: struct
269  fields:
270    name:
271      type: str
272    arg:
273      type: Argument
274Node:
275  kind: struct
276  fields:
277    target:
278      type: str
279    inputs:
280      type: List[NamedArgument]
281    outputs:
282      type: List[Argument]
283    metadata:
284      type: Dict[str, str]
285OptionalTensorArgument:
286  kind: union
287  fields:
288    as_tensor:
289      type: TensorArgument
290    as_none:
291      type: Tuple[()]
292OutputSpec:
293  kind: union
294  fields:
295    user_output:
296      type: UserOutputSpec
297    loss_output:
298      type: LossOutputSpec
299    buffer_mutation:
300      type: BufferMutationSpec
301    gradient_to_parameter:
302      type: GradientToParameterSpec
303    gradient_to_user_input:
304      type: GradientToUserInputSpec
305    user_input_mutation:
306      type: UserInputMutationSpec
307    token:
308      type: OutputTokenSpec
309OutputTokenSpec:
310  kind: struct
311  fields:
312    arg:
313      type: TokenArgument
314RangeConstraint:
315  kind: struct
316  fields:
317    min_val:
318      type: int
319    max_val:
320      type: int
321ScalarType:
322  kind: enum
323  fields:
324    UNKNOWN: 0
325    BYTE: 1
326    CHAR: 2
327    SHORT: 3
328    INT: 4
329    LONG: 5
330    HALF: 6
331    FLOAT: 7
332    DOUBLE: 8
333    COMPLEXHALF: 9
334    COMPLEXFLOAT: 10
335    COMPLEXDOUBLE: 11
336    BOOL: 12
337    BFLOAT16: 13
338SchemaVersion:
339  kind: struct
340  fields:
341    major:
342      type: int
343    minor:
344      type: int
345SymBool:
346  kind: union
347  fields:
348    as_expr:
349      type: SymExpr
350    as_bool:
351      type: bool
352SymBoolArgument:
353  kind: union
354  fields:
355    as_name:
356      type: str
357    as_bool:
358      type: bool
359SymExpr:
360  kind: struct
361  fields:
362    expr_str:
363      type: str
364    hint:
365      type: Optional[SymExprHint]
366      default: None
367SymExprHint:
368  kind: union
369  fields:
370    as_int:
371      type: int
372    as_float:
373      type: float
374    as_bool:
375      type: bool
376SymInt:
377  kind: union
378  fields:
379    as_expr:
380      type: SymExpr
381    as_int:
382      type: int
383SymIntArgument:
384  kind: union
385  fields:
386    as_name:
387      type: str
388    as_int:
389      type: int
390TensorArgument:
391  kind: struct
392  fields:
393    name:
394      type: str
395TensorMeta:
396  kind: struct
397  fields:
398    dtype:
399      type: ScalarType
400    sizes:
401      type: List[SymInt]
402    requires_grad:
403      type: bool
404    device:
405      type: Device
406    strides:
407      type: List[SymInt]
408    storage_offset:
409      type: SymInt
410    layout:
411      type: Layout
412TokenArgument:
413  kind: struct
414  fields:
415    name:
416      type: str
417UserInputMutationSpec:
418  kind: struct
419  fields:
420    arg:
421      type: TensorArgument
422    user_input_name:
423      type: str
424UserInputSpec:
425  kind: struct
426  fields:
427    arg:
428      type: Argument
429UserOutputSpec:
430  kind: struct
431  fields:
432    arg:
433      type: Argument
434SCHEMA_VERSION:
435- 7
436- 3
437TREESPEC_VERSION: 1
438