Searched refs:rendezvous_key_base (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 193 int device_ordinal, string rendezvous_key_base) { in RunExecutable() argument
218 std::move(arguments), rendezvous_key_base, rng_seed, in RunExecutable()
235 int device_ordinal, string rendezvous_key_base) { in ExecuteTPUProgram() argument
241 device_ordinal, rendezvous_key_base); in ExecuteTPUProgram()
307 string rendezvous_key_base = std::to_string(compilation_handle); in DoWork() local
360 core_index, device_ordinal, rendezvous_key_base)); in DoWork()
434 string rendezvous_key_base = std::to_string(op.computation_handle()); in DoWork() local
461 device_ordinal, rendezvous_key_base)); in DoWork()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 109 const string rendezvous_key_base = input.vec<tstring>()(1); in ComputeAsync() local
131 strings::StrCat(rendezvous_key_base, key_, "_dtoh_", i), in ComputeAsync()
264 const string rendezvous_key_base = key_input.vec<tstring>()(1); in Compute() local
278 strings::StrCat(rendezvous_key_base, key_, "_htod_", i), in Compute()
|
| D | tpu_execute_op.cc | 69 OpKernelContext* context, string* rendezvous_key_base, in GetComputationCacheEntry() argument
87 *rendezvous_key_base = key->vec<tstring>()(1); in GetComputationCacheEntry()
641 string rendezvous_key_base; in DoWork() local
644 GetComputationCacheEntry(context, &rendezvous_key_base, &entry_ref)); in DoWork()
758 rendezvous_key_base, rng_seed, node_context.get(), in DoWork()
|
| D | tpu_compile_op_common.cc | 785 std::string rendezvous_key_base = strings::StrCat( in ComputeInternal() local
819 output.vec<tstring>()(1) = rendezvous_key_base; in ComputeInternal()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 75 const std::string& rendezvous_key_base, OpKernelContext* ctx);
83 const string& rendezvous_key_base, in Initialize() argument
417 const string& rendezvous_key_base, uint32 rng_seed, in TPUExecute() argument
435 host_transfers, rendezvous_key_base, ctx)); in TPUExecute()
|
| D | tpu_execute.h | 45 const std::string& rendezvous_key_base, uint32 rng_seed,
|