| 68 | |
| 69 | @dataclass |
| 70 | class Tensor: |
| 71 | scalar_type: ScalarType |
| 72 | storage_offset: int |
| 73 | sizes: List[int] |
| 74 | dim_order: List[int] |
| 75 | requires_grad: bool |
| 76 | layout: int |
| 77 | data_buffer_idx: int |
| 78 | allocation_info: Optional[AllocationDetails] |
| 79 | |
| 80 | # check program.fbs for explanations. |
| 81 | shape_dynamism: TensorShapeDynamism |
| 82 | extra_tensor_info: Optional[ExtraTensorInfo] = None |
| 83 | |
| 84 | |
| 85 | @dataclass |
no outgoing calls