Function SetDefaultLayoutOnProto

tensorflow/compiler/xla/literal_test.cc:1412–1422

Sets the layout of the given ShapeProto to the default.

// Sets the layout of the given ShapeProto to the default.
void SetDefaultLayoutOnProto(ShapeProto* shape_proto) {
  CHECK(ShapeUtil::IsArrayPrimitiveType(shape_proto->element_type()));
  shape_proto->mutable_layout()->set_format(DENSE);
  auto* minor_to_major =
      shape_proto->mutable_layout()->mutable_minor_to_major();
  minor_to_major->Resize(shape_proto->dimensions_size(), 0);
  const int64 size = minor_to_major->size();
  for (int64 i = 0; i < size; ++i) {
    minor_to_major->Set(i, size - 1 - i);
  }
}

TEST_F(LiteralUtilTest, CopyFromProto_Bool) {
  LiteralProto p;