Home
last modified time | relevance | path

Searched refs:DeviceCheckType (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torchgen/dest/
Dregister_dispatch_key.py30 DeviceCheckType,
270 type: DeviceCheckType, args: list[Argument], method_name: str argument
272 if type == DeviceCheckType.NoCheck:
/external/pytorch/torchgen/
Dnative_function_generation.py19 DeviceCheckType,
354 device_check=DeviceCheckType.NoCheck,
Dmodel.py449 class DeviceCheckType(Enum): class
495 device_check: DeviceCheckType
648 device_check_s is None or device_check_s in DeviceCheckType.__members__
650 device_check: DeviceCheckType
652 device_check = DeviceCheckType.ExactSame
654 device_check = DeviceCheckType[device_check_s]
1013 assert self.device_check == DeviceCheckType.NoCheck, (