static constexpr int32_t kArg4Shapes[] = {
5
  };
  static const ShapeInfo* ArgShapeInfos() {
    static constexpr ShapeInfo kArgShapeInfoTable[kNumArgs] = {
{ kArg0Shapes, 2 },
{ kArg1Shapes, 2 },
{ kArg2Shapes, 1 },
{ kArg3Shapes, 1 },
{ kArg4Shapes, 1 },
    };
    return kArgShapeInfoTable;
  };

  // Shapes of the results.
  static constexpr int32_t kResult0Shapes[] = {
5, 6
  };

    static constexpr ShapeInfo kArgShapeInfoTable[kNumArgs] = {
)";
  for (int i = 0; i < ps.parameters_size(); ++i) {
    const xla::ShapeProto& shape = ps.parameters(i);
    *infos +=
        absl::Substitute("{ kArg$0Shapes, $1 },\n", i, shape.dimensions_size());
  }
  *infos += R"(    };
    return kArgShapeInfoTable;
  })";
  return absl::OkStatus();
}