[XLA] Fix broken HLO test

PiperOrigin-RevId: 352813527 Change-Id: Ia605e5ce9f033b397df202577fc403c5ba022d3a
2021-01-20 09:52:25 -08:00 · 2021-01-20 09:52:25 -08:00 · b64b75c35b
commit b64b75c35b
parent a801850ccc
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/compiler/xla/tests/collective_ops_test.cc
+++ b/tensorflow/compiler/xla/tests/collective_ops_test.cc
@ -770,7 +770,7 @@ XLA_TEST_F(CollectiveOpsTest, AllGather_Dim1) {
  ENTRY test_computation {
    id = u32[] replica-id()
    id2 = u32[2, 1] broadcast(id), dimensions={}
-    a0 = u32[2, 1] constant({{10, 15}})
+    a0 = u32[2, 1] constant({10}, {15})
    a1 = u32[2, 1] add(id2, a0)
    allgather = u32[2, 4] all-gather(a1), dimensions={1}
    ROOT out = u32[8] reshape(allgather)