docs: public void addWeightParam(@NonNull String paramKey, @NonNull long... paramShape) {
          Preconditions.checkArgument(paramShape.length > 0, "Provided weight parameter shape is"
                          + " invalid: length 0 provided for shape. Parameter: " + paramKey);
          weightParams.put(paramKey, paramShape);
          paramsList = null;
          weightParamsList = null;
          biasParamsList = null;
      }