Home
last modified time | relevance | path

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

/external/pytorch/torch/_inductor/
Dremote_cache.py33 class RemoteCacheBackend(Generic[_T]): class
81 backend_override_cls: Optional[Callable[[], RemoteCacheBackend[Any]]] = None
84 self, backend: RemoteCacheBackend[_U], serde: RemoteCacheSerde[_T, _U]
126 class RedisRemoteCacheBackend(RemoteCacheBackend[bytes]):
186 backend = typing.cast(RemoteCacheBackend[bytes], None)
/external/pytorch/test/inductor/
Dmock_cache.py14 from torch._inductor.remote_cache import RemoteCacheBackend
79 class MockBackend(RemoteCacheBackend[Any]):
/external/pytorch/torch/_inductor/runtime/
Dautotune_cache.py17 RemoteCacheBackend,
225 class _LocalAutotuneCacheBackend(RemoteCacheBackend[bytes]):