Searched refs:replica_groups_str (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.h | 175 const void* replica_groups_str, xla::int32 replica_groups_str_size,
188 xla::int64 op_id, const void* replica_groups_str,
|
| D | cpu_runtime.cc | 620 xla::int64 op_id, const void* replica_groups_str, in __xla_cpu_runtime_AllToAll() argument
629 static_cast<const char*>(replica_groups_str), replica_groups_str_size); in __xla_cpu_runtime_AllToAll()
661 const void* replica_groups_str, xla::int32 replica_groups_str_size, in __xla_cpu_runtime_AllReduce() argument
667 static_cast<const char*>(replica_groups_str), replica_groups_str_size); in __xla_cpu_runtime_AllReduce()
|