| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compilation_cache_interface.cc | 26 TpuCompilationCacheInterface::RefHolder::RefHolder( in RefHolder()
27 TpuCompilationCacheInterface* parent) in RefHolder()
33 TpuCompilationCacheInterface::RefHolder::~RefHolder() { in ~RefHolder()
39 void TpuCompilationCacheInterface::RefHolder::AddRef(CompiledSubgraph* entry) { in AddRef()
43 std::string TpuCompilationCacheInterface::RefHolder::DebugString() const { in DebugString()
51 TpuCompilationCacheInterface* parent, CompiledSubgraph* entry, int index) in CompilationCacheEntryRef()
114 TpuCompilationCacheInterface::TpuCompilationCacheInterface( in TpuCompilationCacheInterface() function in tensorflow::tpu::TpuCompilationCacheInterface
121 TpuCompilationCacheInterface::~TpuCompilationCacheInterface() { in ~TpuCompilationCacheInterface()
145 Status TpuCompilationCacheInterface::MarkEntryForEviction( in MarkEntryForEviction()
193 Status TpuCompilationCacheInterface::Release(int64_t subgraph_uid) { in Release()
[all …]
|
| H A D | tpu_compilation_cache_interface.h | 58 CompilationCacheEntryRef(TpuCompilationCacheInterface* parent,
78 TpuCompilationCacheInterface* parent_; // Not owned.
86 class TpuCompilationCacheInterface : public ResourceBase {
88 explicit TpuCompilationCacheInterface(int64_t max_cache_size);
89 ~TpuCompilationCacheInterface() override;
177 explicit RefHolder(TpuCompilationCacheInterface* parent);
188 TpuCompilationCacheInterface* parent_; // Not owned.
321 TpuCompilationCacheInterface(const TpuCompilationCacheInterface&) = delete;
322 TpuCompilationCacheInterface& operator=(const TpuCompilationCacheInterface&) =
|
| H A D | tpu_compilation_cache_factory.cc | 28 TpuCompilationCacheInterface* CreateCompilationCacheExternal() { in CreateCompilationCacheExternal()
36 static std::function<TpuCompilationCacheInterface*()>*
38 new std::function<TpuCompilationCacheInterface*()>(
43 std::function<TpuCompilationCacheInterface*()> GetCompilationCacheCreateFn() { in GetCompilationCacheCreateFn()
48 std::function<TpuCompilationCacheInterface*()> fn) { in SetCompilationCacheCreateFn()
51 new std::function<TpuCompilationCacheInterface*()>(fn); in SetCompilationCacheCreateFn()
|
| H A D | tpu_configuration_ops.cc | 97 ResourceMgr* rmgr, tpu::TpuCompilationCacheInterface** compilation_cache) { in CreateTpuCompilationCache()
98 return rmgr->LookupOrCreate<tpu::TpuCompilationCacheInterface>( in CreateTpuCompilationCache()
100 compilation_cache, [&](tpu::TpuCompilationCacheInterface** new_cache) { in CreateTpuCompilationCache()
142 tpu::TpuCompilationCacheInterface* compilation_cache; in Compute()
267 OP_REQUIRES_OK(ctx, DeleteIfExists<tpu::TpuCompilationCacheInterface>( in Compute()
306 tpu::TpuCompilationCacheInterface* compilation_cache; in Compute()
314 tpu::TpuCompilationCacheInterface* local_compilation_cache; in Compute()
|
| H A D | tpu_compilation_cache_external.h | 46 class TpuCompilationCacheExternal : public TpuCompilationCacheInterface {
49 : TpuCompilationCacheInterface(max_cache_size) {} in TpuCompilationCacheExternal()
65 ABSL_EXCLUSIVE_LOCKS_REQUIRED(TpuCompilationCacheInterface::mu_) override;
|
| H A D | tpu_compilation_cache_factory.h | 25 std::function<TpuCompilationCacheInterface*()> GetCompilationCacheCreateFn();
28 std::function<TpuCompilationCacheInterface*()> fn);
|
| H A D | tpu_compilation_cache_local_lookup.h | 32 explicit TpuCompilationCacheLocalLookup(TpuCompilationCacheInterface* cache);
48 TpuCompilationCacheInterface* cache_;
|
| H A D | tpu_compilation_cache_entry_unloader.h | 28 explicit TpuCompilationCacheEntryUnloader(TpuCompilationCacheInterface* cache) in TpuCompilationCacheEntryUnloader()
62 TpuCompilationCacheInterface* cache_; // Not owned.
|
| H A D | compiled_subgraph.h | 29 class TpuCompilationCacheInterface; variable
106 TpuCompilationCacheInterface* parent = nullptr; // Not owned.
|
| H A D | tpu_compilation_cache_service.h | 40 tpu::TpuCompilationCacheInterface* cache);
60 tpu::TpuCompilationCacheInterface* cache_;
|
| H A D | tpu_handle_to_key_op.cc | 40 tpu::TpuCompilationCacheInterface* cache; in Compute()
41 OP_REQUIRES_OK(ctx, rm->Lookup<tpu::TpuCompilationCacheInterface>( in Compute()
|
| H A D | tpu_compilation_cache_service.cc | 28 using ::tensorflow::tpu::TpuCompilationCacheInterface;
34 ::grpc::ServerBuilder* server_builder, TpuCompilationCacheInterface* cache) in TpuCompilationCacheService()
|
| H A D | tpu_pod_state.cc | 56 tpu::TpuCompilationCacheInterface* compilation_cache) { in ConstructCacheService()
148 tpu::TpuCompilationCacheInterface* compilation_cache, in ConstructTpuPodState()
|
| H A D | tpu_compile_op_common.cc | 255 TpuCompilationCacheInterface* cache; in ComputeInternal()
256 TF_RETURN_IF_ERROR(rm->Lookup<TpuCompilationCacheInterface>( in ComputeInternal()
|
| H A D | tpu_pod_state.h | 57 tpu::TpuCompilationCacheInterface* compilation_cache,
|
| H A D | tpu_compilation_cache_local_lookup.cc | 21 TpuCompilationCacheInterface* cache) in TpuCompilationCacheLocalLookup()
|
| H A D | tpu_configuration_ops.h | 29 ResourceMgr* rmgr, tpu::TpuCompilationCacheInterface** compilation_cache);
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | tpu_compile_ops.cc | 187 tpu::TpuCompilationCacheInterface* cache; in Compute()
188 OP_REQUIRES_OK(ctx, rm->Lookup<tpu::TpuCompilationCacheInterface>( in Compute()
248 tpu::TpuCompilationCacheInterface* cache; in Compute()
249 OP_REQUIRES_OK(ctx, rm->Lookup<tpu::TpuCompilationCacheInterface>( in Compute()
|
| H A D | tpu_execute_op.cc | 322 tpu::TpuCompilationCacheInterface* cache; in DoWork()
323 TF_RETURN_IF_ERROR(rm->Lookup<tpu::TpuCompilationCacheInterface>( in DoWork()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_tpu_kernels.cc | 135 tpu::TpuCompilationCacheInterface* compilation_cache; in InitializeInternal()
217 tpu::TpuCompilationCacheInterface* local_compilation_cache; in InitializeInternal()
|