Searched refs:channel_id_present (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.h | 176 xla::int32 channel_id_present, xla::int64 op_id, xla::int32 reduction_kind,
181 const xla::ExecutableRunOptions* run_options, xla::int32 channel_id_present,
187 const xla::ExecutableRunOptions* run_options, xla::int32 channel_id_present,
|
| D | cpu_runtime.cc | 597 std::vector<xla::ReplicaGroup> group, xla::int32 channel_id_present, in GetRendezvousKey() argument
603 channel_id_present ? xla::RendezvousKey::kCrossModule in GetRendezvousKey()
619 const xla::ExecutableRunOptions* run_options, xla::int32 channel_id_present, in __xla_cpu_runtime_AllToAll() argument
633 GetRendezvousKey(run_options, group, channel_id_present, op_id); in __xla_cpu_runtime_AllToAll()
662 xla::int32 channel_id_present, xla::int64 op_id, xla::int32 reduction_kind, in __xla_cpu_runtime_AllReduce() argument
671 GetRendezvousKey(run_options, group, channel_id_present, op_id); in __xla_cpu_runtime_AllReduce()
720 const xla::ExecutableRunOptions* run_options, xla::int32 channel_id_present, in __xla_cpu_runtime_CollectivePermute() argument
743 GetRendezvousKey(run_options, {}, channel_id_present, op_id); in __xla_cpu_runtime_CollectivePermute()
|