| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compilation_cache_interface.cc | 47 CompilationCacheEntryRef::CompilationCacheEntryRef() in CompilationCacheEntryRef() function in tensorflow::tpu::CompilationCacheEntryRef
50 CompilationCacheEntryRef::CompilationCacheEntryRef( in CompilationCacheEntryRef() function in tensorflow::tpu::CompilationCacheEntryRef
65 CompilationCacheEntryRef::~CompilationCacheEntryRef() { in ~CompilationCacheEntryRef()
76 TpuCompilationCacheEntry CompilationCacheEntryRef::get() { in get()
86 Status CompilationCacheEntryRef::ToSubEntryRef( in ToSubEntryRef()
562 std::unique_ptr<CompilationCacheEntryRef>* entry) { in Lookup()
580 *entry = absl::make_unique<CompilationCacheEntryRef>(this, cache_entry, in Lookup()
587 std::unique_ptr<CompilationCacheEntryRef>* entry) { in Lookup()
600 *entry = absl::make_unique<CompilationCacheEntryRef>(this, cache_entry, in Lookup()
|
| H A D | tpu_compilation_cache_interface.h | 55 class CompilationCacheEntryRef {
57 CompilationCacheEntryRef();
58 CompilationCacheEntryRef(TpuCompilationCacheInterface* parent,
61 virtual ~CompilationCacheEntryRef();
141 std::unique_ptr<CompilationCacheEntryRef>* entry);
147 std::unique_ptr<CompilationCacheEntryRef>* entry);
|
| H A D | tpu_compilation_cache_lookup.h | 47 std::unique_ptr<CompilationCacheEntryRef>* entry,
51 std::unique_ptr<CompilationCacheEntryRef>* entry) { in Lookup()
61 std::unique_ptr<CompilationCacheEntryRef>* entry,
65 std::unique_ptr<CompilationCacheEntryRef>* entry) { in Lookup()
|
| H A D | tpu_compilation_cache_rpc_lookup.cc | 65 std::unique_ptr<CompilationCacheEntryRef>* entry, in Lookup()
100 std::unique_ptr<CompilationCacheEntryRef>* entry, in Lookup()
172 std::unique_ptr<CompilationCacheEntryRef>* entry, in PostLookupLocked()
177 std::unique_ptr<CompilationCacheEntryRef>(new CacheWrapper(*cache_entry)); in PostLookupLocked()
|
| H A D | tpu_compilation_cache_rpc_lookup.h | 45 std::unique_ptr<tpu::CompilationCacheEntryRef>* entry,
49 std::unique_ptr<tpu::CompilationCacheEntryRef>* entry,
66 std::unique_ptr<tpu::CompilationCacheEntryRef>* entry,
|
| H A D | tpu_compilation_cache_local_lookup.h | 36 std::unique_ptr<CompilationCacheEntryRef>* entry,
40 std::unique_ptr<CompilationCacheEntryRef>* entry,
|
| H A D | tpu_compilation_cache_local_lookup.cc | 31 const string& proto_key, std::unique_ptr<CompilationCacheEntryRef>* entry, in Lookup()
50 std::unique_ptr<CompilationCacheEntryRef>* entry, in Lookup()
|
| H A D | tpu_compilation_cache_service.cc | 26 using ::tensorflow::tpu::CompilationCacheEntryRef;
90 std::unique_ptr<CompilationCacheEntryRef> entry; in GetTpuProgram()
|
| H A D | tpu_execute_op.cc | 62 using ::tensorflow::tpu::CompilationCacheEntryRef;
70 std::unique_ptr<CompilationCacheEntryRef>* entry) { in GetComputationCacheEntry()
637 std::unique_ptr<CompilationCacheEntryRef> entry_ref; in DoWork()
|
| H A D | tpu_compilation_cache_rpc_support.h | 56 class CacheWrapper : public CompilationCacheEntryRef {
|
| H A D | tpu_reshard_variables_op_util.h | 40 std::unique_ptr<tpu::CompilationCacheEntryRef>* entry,
|
| H A D | tpu_reshard_variables_op.cc | 143 std::unique_ptr<tpu::CompilationCacheEntryRef> entry_ref; in DoTpuExecute()
|
| H A D | tpu_reshard_variables_op_util.cc | 80 std::unique_ptr<tpu::CompilationCacheEntryRef>* entry, in GetComputationCacheEntry()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | tpu_execute_op.cc | 54 using tensorflow::tpu::CompilationCacheEntryRef;
63 std::unique_ptr<CompilationCacheEntryRef>* entry) { in GetComputationCacheEntry()
310 std::unique_ptr<CompilationCacheEntryRef> entry; in DoWork()
433 std::unique_ptr<CompilationCacheEntryRef> entry; in DoWork()
|