Searched refs:GetRendezvousKey (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sendrecv_ops.cc | 38 static void GetRendezvousKey(const string& key_prefix, in GetRendezvousKey() function
73 GetRendezvousKey(key_prefix_, {0, 0}, &parsed_key_.buf_); in SendOp()
103 GetRendezvousKey(key_prefix_, frame_iter, &in_loop_parsed.buf_); in Compute()
152 GetRendezvousKey(key_prefix_, {0, 0}, &parsed_key_.buf_); in RecvOp()
209 GetRendezvousKey(key_prefix_, frame_iter, &in_loop_parsed.buf_); in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | kernel_benchmark_testlib.h | 104 string GetRendezvousKey(const Node* node);
|
| D | executor_test.cc | 524 string x_key = test::GetRendezvousKey(x); in BM_FeedInputFetchOutput()
525 string y_key = test::GetRendezvousKey(y); in BM_FeedInputFetchOutput()
526 string z_key = test::GetRendezvousKey(z); in BM_FeedInputFetchOutput()
|
| D | kernel_benchmark_testlib.cc | 147 string GetRendezvousKey(const Node* node) { in GetRendezvousKey() function
|
| D | direct_session.cc | 147 string GetRendezvousKey(const string& tensor_name, in GetRendezvousKey() function
1448 ek->input_name_to_rendezvous_key[input] = GetRendezvousKey( in CreateExecutors()
1454 GetRendezvousKey(output, device_set_.client_device()->attributes(), in CreateExecutors()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 595 xla::RendezvousKey GetRendezvousKey( in GetRendezvousKey() function
633 GetRendezvousKey(run_options, group, channel_id_present, op_id); in __xla_cpu_runtime_AllToAll()
671 GetRendezvousKey(run_options, group, channel_id_present, op_id); in __xla_cpu_runtime_AllReduce()
743 GetRendezvousKey(run_options, {}, channel_id_present, op_id); in __xla_cpu_runtime_CollectivePermute()
|