Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A 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)
/aosp_15_r20/external/pytorch/test/inductor/
H A Dmock_cache.py14 from torch._inductor.remote_cache import RemoteCacheBackend
79 class MockBackend(RemoteCacheBackend[Any]):
/aosp_15_r20/external/pytorch/torch/_inductor/runtime/
H A Dautotune_cache.py17 RemoteCacheBackend,
225 class _LocalAutotuneCacheBackend(RemoteCacheBackend[bytes]):