1
  };
  static constexpr int32_t kArg3Shapes[] = {
1
  };
  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.