Searched refs:local_proto_key (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compilation_cache_rpc_lookup.cc | 75 std::string local_proto_key = absl::StrCat( in Lookup() local
80 auto iter = cache_.find(local_proto_key); in Lookup()
86 RemoteLookupLocked(local_proto_key, request, &cache_entry)); in Lookup()
88 VLOG(1) << "Found key " << local_proto_key << " in local proto cache."; in Lookup()
116 std::string local_proto_key = in Lookup() local
121 auto iter = cache_.find(local_proto_key); in Lookup()
130 RemoteLookupLocked(local_proto_key, request, &cache_entry)); in Lookup()
144 const std::string& local_proto_key, in RemoteLookupLocked() argument
158 VLOG(1) << "Looked up key " << local_proto_key in RemoteLookupLocked()
163 local_proto_key, &response, cache_entry)); in RemoteLookupLocked()
[all …]
|
| D | tpu_compilation_cache_rpc_support.cc | 36 absl::string_view local_proto_key, GetTpuProgramResponseExternal* response, in DeserializeRpcResponseToCacheEntry() argument
42 entry.key = std::string(local_proto_key); in DeserializeRpcResponseToCacheEntry()
|
| D | tpu_compilation_cache_rpc_support.h | 87 const absl::string_view local_proto_key, ResponseType* response,
|
| D | tpu_compilation_cache_rpc_lookup.h | 56 Status RemoteLookupLocked(const string& local_proto_key,
|