Searched refs:shape_copy (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_test.cc | 53 Shape shape_copy(shape.ToProto()); in TEST_F() local
54 EXPECT_TRUE(ShapeUtil::Equal(shape, shape_copy)) in TEST_F()
55 << shape << " != " << shape_copy; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 948 Shape shape_copy = ShapeUtil::MakeShapeWithLayout(F32, {4, 5}, {1, 0}); in TEST_F() local
953 m::Slice(m::Copy(m::Parameter(1)).WithShapeEqualTo(&shape_copy))))); in TEST_F()
980 Shape shape_copy = ShapeUtil::MakeShapeWithLayout(F32, {4, 5}, {1, 0}); in TEST_F() local
985 m::Copy(m::Parameter(1)).WithShapeEqualTo(&shape_copy), in TEST_F()
1013 Shape shape_copy = ShapeUtil::MakeShapeWithLayout(F32, {3, 5}, {1, 0}); in TEST_F() local
1018 m::Concatenate(m::Copy(m::Parameter(1)).WithShapeEqualTo(&shape_copy), in TEST_F()
1070 Shape shape_copy = ShapeUtil::MakeShapeWithLayout(F32, {4, 5}, {0, 1}); in TEST_F() local
1073 m::Copy(m::Parameter(0)).WithShapeEqualTo(&shape_copy)))); in TEST_F()
|