Searched refs:shape_copy (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_test.cc | 56 Shape shape_copy(shape.ToProto()); in TEST_F() local
57 EXPECT_TRUE(ShapeUtil::Equal(shape, shape_copy)) in TEST_F()
58 << shape << " != " << shape_copy; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 942 Shape shape_copy = ShapeUtil::MakeShapeWithLayout(F32, {4, 5}, {1, 0}); in TEST_F() local
947 m::Slice(m::Copy(m::Parameter(1)).WithShapeEqualTo(&shape_copy))))); in TEST_F()
974 Shape shape_copy = ShapeUtil::MakeShapeWithLayout(F32, {4, 5}, {1, 0}); in TEST_F() local
979 m::Copy(m::Parameter(1)).WithShapeEqualTo(&shape_copy), in TEST_F()
1007 Shape shape_copy = ShapeUtil::MakeShapeWithLayout(F32, {3, 5}, {1, 0}); in TEST_F() local
1012 m::Concatenate(m::Copy(m::Parameter(1)).WithShapeEqualTo(&shape_copy), in TEST_F()
1064 Shape shape_copy = ShapeUtil::MakeShapeWithLayout(F32, {4, 5}, {0, 1}); in TEST_F() local
1067 m::Copy(m::Parameter(0)).WithShapeEqualTo(&shape_copy)))); in TEST_F()
|