From b64b75c35b61b8c671e55800db98f8df75d73d91 Mon Sep 17 00:00:00 2001
From: David Majnemer <majnemer@google.com>
Date: Wed, 20 Jan 2021 09:52:25 -0800
Subject: [PATCH] [XLA] Fix broken HLO test

PiperOrigin-RevId: 352813527
Change-Id: Ia605e5ce9f033b397df202577fc403c5ba022d3a
---
 tensorflow/compiler/xla/tests/collective_ops_test.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/compiler/xla/tests/collective_ops_test.cc b/tensorflow/compiler/xla/tests/collective_ops_test.cc
index 747dfa73151..32159e16d75 100644
--- 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)