| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compilation_cache_lookup.h | 46 virtual Status Lookup(const string& proto_key,
50 virtual Status Lookup(const string& proto_key, in Lookup() argument
52 return Lookup(proto_key, std::move(entry), in Lookup()
|
| D | tpu_compilation_cache_local_lookup.cc | 31 const string& proto_key, std::unique_ptr<CompilationCacheEntryRef>* entry, in Lookup() argument
35 Status s = cache_->Lookup(proto_key, entry); in Lookup()
36 VLOG(1) << "Looked up key " << proto_key << " in local subgraph cache status " in Lookup()
|
| D | tpu_compilation_cache_interface.cc | 266 for (const std::string& key : entry->proto_key) { in DiscardEntryRef()
365 std::vector<std::string>* proto_key, std::vector<std::string>* sharding_key, in CompileIfKeyAbsent() argument
371 subgraph_key, session_metadata, per_step_ref_holder, uid, proto_key, in CompileIfKeyAbsent()
403 std::vector<std::string>* proto_key, std::vector<std::string>* sharding_key, in CompileIfKeyAbsentHelper() argument
502 *proto_key = entry->proto_key; in CompileIfKeyAbsentHelper()
552 *keys = iter->second->proto_key; in GetKeysFromUid()
582 const std::string& proto_key, in Lookup() argument
590 const auto iter = entries_by_proto_key_.find(proto_key); in Lookup()
592 return errors::NotFound("No proto found for key ", proto_key); in Lookup()
|
| D | tpu_compilation_cache_external.cc | 47 entry->proto_key.push_back(ProtoKeyForComputation(key, i)); in PopulateEntry()
135 for (int64 i = 0; i < main_entry->proto_key.size(); ++i) { in InitializeEntry()
138 main_entry->proto_key[i], std::make_pair(main_entry, i))); in InitializeEntry()
|
| D | tpu_compilation_cache_interface.h | 111 std::vector<std::string>* proto_key,
140 Status Lookup(const std::string& proto_key,
200 std::vector<std::string>* proto_key,
|
| D | tpu_compilation_cache_rpc_lookup.cc | 64 const std::string& proto_key, in Lookup() argument
76 proto_key, "_", tpu::CompilationCacheFetchTarget_Name(fetch_target)); in Lookup()
83 request.set_key(proto_key); in Lookup()
|
| D | tpu_compile_op_common.cc | 688 std::vector<std::string> proto_key; in ComputeInternal() local
693 key, ctx->session_metadata(), ref_holder, &uid, &proto_key, &sharding_key, in ComputeInternal()
759 int64 num_cores_with_compiled_programs = proto_key.size(); in ComputeInternal()
760 if (proto_key.size() == 1) { in ComputeInternal()
814 if (proto_key.size() == 1) { in ComputeInternal()
815 output.vec<tstring>()(0) = proto_key[0]; in ComputeInternal()
817 output.vec<tstring>()(0) = proto_key[i]; in ComputeInternal()
|
| D | tpu_compilation_cache_local_lookup.h | 35 Status Lookup(const string& proto_key,
|
| D | compiled_subgraph.h | 125 std::vector<std::string> proto_key; member
|
| D | tpu_compilation_cache_rpc_lookup.h | 44 Status Lookup(const string& proto_key,
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_compile_ops.cc | 193 std::vector<string> proto_key; in Compute() local
200 /*per_step_ref_holder=*/nullptr, &uid, &proto_key, &shard_key, in Compute()
|
| /external/protobuf/php/src/Google/Protobuf/Internal/ |
| D | Message.php | 1191 $proto_key = $this->convertJsonValueToProtoValue(
1198 self::kvUpdateHelper($field, $proto_key, $proto_value);
|