Searched refs:SchemaCheckMode (Results 1 – 7 of 7) sorted by relevance
11 from torch._subclasses.schema_check_mode import SchemaCheckMode104 with SchemaCheckMode() as schema_check:111 with SchemaCheckMode() as schema_check:121 with SchemaCheckMode() as schema_check:129 with SchemaCheckMode() as schema_check:137 with SchemaCheckMode() as schema_check:146 with SchemaCheckMode() as schema_check:166 with SchemaCheckMode() as schema_check:186 with SchemaCheckMode() as schema_check:207 with SchemaCheckMode() as schema_check:[all …]
5 from torch._subclasses.schema_check_mode import SchemaCheckMode24 class PreDispatchSchemaCheckMode(SchemaCheckMode):
73 from torch._subclasses.schema_check_mode import SchemaCheckMode79 with SchemaCheckMode():
235 from torch._subclasses.schema_check_mode import SchemaCheckMode251 if isinstance(i, SchemaCheckMode):270 isinstance(old, SchemaCheckMode)
62 class SchemaCheckMode(TorchDispatchMode): class
541 from torch._subclasses.schema_check_mode import SchemaCheckMode549 SchemaCheckMode,554 if isinstance(mode, SchemaCheckMode):
20 from torch._subclasses.schema_check_mode import SchemaCheckMode57 with SchemaCheckMode():