Searched refs:replica_groups_str_size (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,
189 xla::int32 replica_groups_str_size, xla::int32 num_buffers,
|
| D | cpu_runtime.cc | 621 xla::int32 replica_groups_str_size, xla::int32 num_buffers, 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()
|