tilus.Script.load_shared¶
- Script.load_shared(src, *, layout=None, out=None)[source]¶
Load a shared tensor into a register tensor.
- Parameters:
src (SharedTensor) – The shared tensor to load from.
layout (RegisterLayout, optional) – The layout of the register tensor. If not provided, the layout will be inferred based on the operations performed on it. When provided, its shape must match the shape of the shared tensor.
out (RegisterTensor, optional) – The register tensor to store the loaded data into. If not provided, a new register tensor will be allocated.
- Returns:
ret – The register tensor containing the loaded data from the shared tensor.
- Return type: