};
  static constexpr int32_t kResult2Shapes[] = {
5
  };
  static const ShapeInfo* ResultShapeInfos() {
    static constexpr ShapeInfo kResultShapeInfoTable[kNumResults] = {
{ kResult0Shapes, 2 },
{ kResult1Shapes, 1 },
{ kResult2Shapes, 1 },
    };
    return kResultShapeInfoTable;
  };

  // Array of names of each positional argument, terminated by nullptr.
  static const char** StaticArgNames() {

    static constexpr ShapeInfo kResultShapeInfoTable[kNumResults] = {
)";
  for (int i = 0; i < ps.result().tuple_shapes_size(); ++i) {
    const xla::ShapeProto& shape = ps.result().tuple_shapes(i);
    *infos += absl::Substitute("{ kResult$0Shapes, $1 },\n", i,
                               shape.dimensions_size());
  }
  *infos += R"(    };
    return kResultShapeInfoTable;
  })";
  return absl::OkStatus();
}