AllocationSpec is used to represent the allocation of a tensor.
| 689 | |
| 690 | @dataclass |
| 691 | class AllocationSpec: |
| 692 | """ |
| 693 | AllocationSpec is used to represent the allocation of a tensor. |
| 694 | """ |
| 695 | |
| 696 | # The offset of the tensor in the shared object/pool. |
| 697 | offset: int |
| 698 | # TensorSpec |
| 699 | spec: TensorSpec |
| 700 | |
| 701 | |
| 702 | @dataclass |