Searched defs:rendezvous_key_base (Results 1 – 7 of 7) sorted by relevance
194 int device_ordinal, string rendezvous_key_base) { in RunExecutable()236 int device_ordinal, string rendezvous_key_base) { in ExecuteTPUProgram()309 string rendezvous_key_base = std::to_string(compilation_handle); in DoWork() local436 string rendezvous_key_base = std::to_string(op.computation_handle()); in DoWork() local
109 const string rendezvous_key_base = input.vec<tstring>()(1); in ComputeAsync() local264 const string rendezvous_key_base = key_input.vec<tstring>()(1); in Compute() local
69 OpKernelContext* context, string* rendezvous_key_base, in GetComputationCacheEntry()636 string rendezvous_key_base; in DoWork() local
142 string rendezvous_key_base; in DoTpuExecute() local
79 const Tensor& key, string* rendezvous_key_base, in GetComputationCacheEntry()
386 std::string rendezvous_key_base = strings::StrCat( in ComputeInternal() local
85 const string& rendezvous_key_base, in Initialize()421 const string& rendezvous_key_base, uint32 rng_seed, in TPUExecute()