| 11558 | return { first: r[0], second: r[2] }; |
| 11559 | } |
| 11560 | parseBaseType() { |
| 11561 | const L = this.L; |
| 11562 | const tok = L.cur(); |
| 11563 | const text = tok.text(); |
| 11564 | L.next(); |
| 11565 | switch (text) { |
| 11566 | case 'Tensor': return torch.TensorType.get(); |
| 11567 | case 'bool': return torch.BoolType.get(); |
| 11568 | case 'int': return torch.IntType.get(); |
| 11569 | case 'float': return torch.FloatType.get(); |
| 11570 | case 'complex': return torch.ComplexType.get(); |
| 11571 | case 'str': return torch.StringType.get(); |
| 11572 | case 'SymInt': return torch.SymIntType.get(); |
| 11573 | case 'Scalar': return torch.NumberType.get(); |
| 11574 | case 'ScalarType': return torch.Type.get('ScalarTypeType'); |
| 11575 | case 'Device': return torch.DeviceObjType.get(); |
| 11576 | case 'Layout': return torch.Type.get('Layout'); |
| 11577 | case 'MemoryFormat': return torch.Type.get('MemoryFormat'); |
| 11578 | case 'Generator': return torch._C._GeneratorType.get(); |
| 11579 | case 't': case 't1': case 't2': case 'tVal': return torch._C.VarType.create(text); |
| 11580 | case 'Any': return torch.AnyType.get(); |
| 11581 | case 'AnyEnumType': return torch.Type.get('AnyEnumType'); |
| 11582 | case 'Dimname': return torch.StringType.get(); |
| 11583 | case 'QScheme': return torch.Type.get('QSchemeType'); |
| 11584 | case 'Stream': return torch.StreamObjType.get(); |
| 11585 | case 'Storage': return torch.Type.get('Storage'); |
| 11586 | case 'AnyClassType': return torch.Type.get('AnyClassType'); |
| 11587 | case 'NoneType': return torch.NoneType.get(); |
| 11588 | default: throw new python.Error(`Unsupported type '${text}'.`); |
| 11589 | } |
| 11590 | } |
| 11591 | parseFakeAndRealType() { |
| 11592 | const L = this.L; |
| 11593 | let fake_value = null; |