Searched refs:proto_lookup (Results 1 – 4 of 4) sorted by relevance
65 TpuCompilationCacheLookup* proto_lookup; in GetComputationCacheEntry() local68 &proto_lookup)); in GetComputationCacheEntry()69 core::ScopedUnref lookup_unref(proto_lookup); in GetComputationCacheEntry()70 TF_RETURN_IF_ERROR(proto_lookup->Lookup(key, core_index_in_replica, entry)); in GetComputationCacheEntry()422 TpuCompilationCacheLookup* proto_lookup; in DoWork() local425 &proto_lookup)); in DoWork()426 core::ScopedUnref lookup_unref(proto_lookup); in DoWork()432 TF_RETURN_IF_ERROR(proto_lookup->Lookup( in DoWork()
315 tpu::TpuCompilationCacheLookup* proto_lookup; in Compute() local316 proto_lookup = in Compute()320 tpu::kCompiledProtoCacheResourceName, proto_lookup)); in Compute()350 tpu::TpuCompilationCacheLookup* proto_lookup = in Compute() local354 tpu::kCompiledProtoCacheResourceName, proto_lookup)); in Compute()
81 TpuCompilationCacheLookup* proto_lookup; in GetComputationCacheEntry() local84 &proto_lookup)); in GetComputationCacheEntry()85 core::ScopedUnref lookup_unref(proto_lookup); in GetComputationCacheEntry()86 TF_RETURN_IF_ERROR(proto_lookup->Lookup(key->vec<tstring>()(0), entry)); in GetComputationCacheEntry()
1591 std::map<string, string>::iterator proto_lookup = in AddFile() local1593 if (proto_lookup != proto_file_to_framework_name_.end()) { in AddFile()1595 proto_lookup->second + "/" + in AddFile()