Searched defs:cls_to_become (Results 1 – 7 of 7) sorted by relevance
1221 cls_to_become = Conv1d # type: ignore[assignment] variable in LazyConv1d1290 cls_to_become = Conv2d # type: ignore[assignment] variable in LazyConv2d1360 cls_to_become = Conv3d # type: ignore[assignment] variable in LazyConv3d1427 cls_to_become = ConvTranspose1d # type: ignore[assignment] variable in LazyConvTranspose1d1496 cls_to_become = ConvTranspose2d # type: ignore[assignment] variable in LazyConvTranspose2d1565 cls_to_become = ConvTranspose3d # type: ignore[assignment] variable in LazyConvTranspose3d
231 cls_to_become = InstanceNorm1d # type: ignore[assignment] variable in LazyInstanceNorm1d348 cls_to_become = InstanceNorm2d # type: ignore[assignment] variable in LazyInstanceNorm2d464 cls_to_become = InstanceNorm3d # type: ignore[assignment] variable in LazyInstanceNorm3d
371 cls_to_become = BatchNorm1d # type: ignore[assignment] variable in LazyBatchNorm1d482 cls_to_become = BatchNorm2d # type: ignore[assignment] variable in LazyBatchNorm2d593 cls_to_become = BatchNorm3d # type: ignore[assignment] variable in LazyBatchNorm3d
263 cls_to_become = Linear # type: ignore[assignment] variable in LazyLinear
183 cls_to_become: Optional[Type[Any]] = None variable in LazyModuleMixin
197 cls_to_become = Parameter variable in UninitializedParameter270 cls_to_become = torch.Tensor variable in UninitializedBuffer
574 cls_to_become = MaterializedModule variable in LazyModule