code: @Override
      public void update(NDArrayMessage message) {
          updateStorage.addUpdate(message);
          INDArray arr = message.getArr();
          //of note for ndarrays
          int[] dimensions = message.getDimensions();
          boolean whole = dimensions.length == 1 && dimensions[0] == -1;
  
          if (!whole)
              partialUpdate(arr, ndArrayHolder.get(), message.getIndex(), dimensions);
          else
              update(arr, ndArrayHolder.get());
      }
docs: Do an update based on the ndarray message.
  
  @param message