Files
pytorch/torch/export/custom_obj.py
Angela Yi 92d718aed1 [export] Add lifted constant obj to input (#116985)
Test Plan: wip

Differential Revision: D52556070

Pull Request resolved: https://github.com/pytorch/pytorch/pull/116985
Approved by: https://github.com/suo
2024-01-18 22:10:53 +00:00

16 lines
314 B
Python

from dataclasses import dataclass
from typing import Optional
__all__ = ["ScriptObjectMeta"]
@dataclass
class ScriptObjectMeta:
"""
Metadata which is stored on nodes representing ScriptObjects.
"""
# Key into constants table to retrieve the real ScriptObject.
constant_name: Optional[str]