Home
last modified time | relevance | path

Searched refs:CompiledFunctionCache (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Djax_jit.cc364 class CompiledFunctionCache { class
367 explicit CompiledFunctionCache(int capacity);
419 CompiledFunctionCache::CompiledFunctionCache(int capacity) in CompiledFunctionCache() function in jax::__anon901e1bb70811::CompiledFunctionCache
422 std::shared_ptr<CompiledFunctionCache::Cache> CompiledFunctionCache::Lookup( in Lookup()
460 std::shared_ptr<CompiledFunctionCache> cache);
514 const std::shared_ptr<CompiledFunctionCache>& cache() const { return cache_; } in cache()
560 std::shared_ptr<CompiledFunctionCache> cache_;
563 std::shared_ptr<CompiledFunctionCache::Cache> executables_;
611 std::shared_ptr<CompiledFunctionCache> cache) in CompiledFunction()
1206 std::shared_ptr<CompiledFunctionCache> cache) { in InitializeCompiledFunction()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
H A Djax_jit.pyi26 class CompiledFunctionCache:
57 cache: Optional[CompiledFunctionCache] = ...) -> CompiledFunction: ...