mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Pull Request resolved: https://github.com/pytorch/pytorch/pull/138454 Approved by: https://github.com/eellison
25 lines
500 B
Plaintext
25 lines
500 B
Plaintext
# mypy: ignore-errors
|
|
|
|
import collections
|
|
import types
|
|
from typing import Any, Dict, List, Optional, TYPE_CHECKING
|
|
|
|
import torch
|
|
import torch.fx
|
|
from torch._guards import Source
|
|
|
|
from ..utils import (
|
|
namedtuple_fields,
|
|
odict_values,
|
|
OrderedSet,
|
|
set_example_value,
|
|
)
|
|
from .base import MutableLocal, VariableTracker, VariableTrackerContainer
|
|
|
|
if TYPE_CHECKING:
|
|
from torch._dynamo.codegen import PyCodegen
|
|
|
|
|
|
class BaseListVariable(VariableTrackerContainer):
|
|
our_container = set
|