diff options
author | 2017-01-11 20:09:44 -0800 | |
---|---|---|
committer | 2017-01-11 20:25:47 -0800 | |
commit | 401387665d3fb80ddcb0fb216d780ac04d86933f (patch) | |
tree | ea4ef3f6b4a6d86ad6112143243edb8e1a34e263 /tensorflow/compiler/xla/service/shape_inference_test.cc | |
parent | 873473ef0fff3d5f106f87cbd83048222f1dd93a (diff) |
[XLA] Use `ShapeUtil::PermuteDimensions` in `ShapeInference::InferTransposeShape`
Change: 144282152
Diffstat (limited to 'tensorflow/compiler/xla/service/shape_inference_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/shape_inference_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/shape_inference_test.cc b/tensorflow/compiler/xla/service/shape_inference_test.cc index 10fd4e53c5..5a1ae6b002 100644 --- a/tensorflow/compiler/xla/service/shape_inference_test.cc +++ b/tensorflow/compiler/xla/service/shape_inference_test.cc @@ -1125,8 +1125,8 @@ TEST_F(ShapeInferenceTest, Transpose) { ShapeInference::InferTransposeShape(a_shape, {1, 2, 3, 0}); EXPECT_IS_OK(inferred_shape_and_status); Shape inferred_shape = inferred_shape_and_status.ValueOrDie(); - EXPECT_TRUE(ShapeUtil::Equal(inferred_shape, - ShapeUtil::MakeShape(F32, {3, 4, 5, 2}))); + EXPECT_TRUE(ShapeUtil::Compatible(inferred_shape, + ShapeUtil::MakeShape(F32, {3, 4, 5, 2}))); } } // namespace |