experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[xla::Shape::Shape] Emplace tuple shapes into tuple_shapes_.

Browse Source 
This avoids creating and destroying a temporary `Shape` in `add_tuple_shapes()`, then move-assigning and destroying each tuple-shape in the constructor.

PiperOrigin-RevId: 298703231
Change-Id: I793742c163cf24780b0764aa6c3846715517f99d
This commit is contained in:
Derek Murray 2020-03-03 15:02:57 -08:00 committed by TensorFlower Gardener
parent 3421416220
commit a8ac8f901a
1 changed files with 1 additions and 1 deletions

2
tensorflow/compiler/xla/shape.cc
View File

@ -48,7 +48,7 @@ Shape::Shape(const ShapeProto& shape_proto) {
}
tuple_shapes_.reserve(shape_proto.tuple_shapes_size());
for (const ShapeProto& element_shape : shape_proto.tuple_shapes()) {
*add_tuple_shapes() = Shape(element_shape);
tuple_shapes_.emplace_back(element_shape);
}
if (shape_proto.has_layout()) {
*mutable_layout() = Layout::CreateFromProto(shape_proto.layout());