Commit 8012cf52 authored by Youlong Cheng's avatar Youlong Cheng Committed by TensorFlower Gardener
Browse files

[TF:XLA] Change group_assignment from 1d array attribute to 2d array input... 

[TF:XLA] Change group_assignment from 1d array attribute to 2d array input tensor with shape [num_groups, num_replica_per_group].

PiperOrigin-RevId: 210656091
parent 3d35a071
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment