aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/bfloat16_normalization_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/service/bfloat16_normalization_test.cc')
-rw-r--r--tensorflow/compiler/xla/service/bfloat16_normalization_test.cc2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/bfloat16_normalization_test.cc b/tensorflow/compiler/xla/service/bfloat16_normalization_test.cc
index cef0eba14e..2411fdcb20 100644
--- a/tensorflow/compiler/xla/service/bfloat16_normalization_test.cc
+++ b/tensorflow/compiler/xla/service/bfloat16_normalization_test.cc
@@ -284,7 +284,7 @@ TEST_F(BFloat16NormalizationTest, ResolveMixedPrecisionTupleSort) {
HloInstruction::CreateParameter(1, s32_shape, "value"));
HloInstruction* sort = builder.AddInstruction(HloInstruction::CreateSort(
- ShapeUtil::MakeTupleShape({bf16_shape, s32_shape}), 0, key, value));
+ ShapeUtil::MakeTupleShape({bf16_shape, s32_shape}), 0, key, {value}));
HloInstruction* gte = builder.AddInstruction(
HloInstruction::CreateGetTupleElement(bf16_shape, sort, 0));