Searched defs:replica_groups_str_size (Results 1 – 1 of 1) sorted by relevance
643 xla::int32 replica_groups_str_size, xla::int32 num_buffers, in __xla_cpu_runtime_AllToAll()683 const void* replica_groups_str, xla::int32 replica_groups_str_size, in __xla_cpu_runtime_AllReduce()