code: public static int[] getHWDFromInputType(InputType inputType) {
          int inH;
          int inW;
          int inDepth;
  
          // FIXME: int cast
          if (inputType instanceof InputType.InputTypeConvolutional) {
              InputType.InputTypeConvolutional conv = (InputType.InputTypeConvolutional) inputType;
              inH = (int) conv.getHeight();
              inW = (int) conv.getWidth();
              inDepth = (int) conv.getChannels();
          } else if (inputType instanceof InputType.InputTypeConvolutionalFlat) {
              InputType.InputTypeConvolutionalFlat conv = (InputType.InputTypeConvolutionalFlat) inputType;
              inH = (int) conv.getHeight();
              inW = (int) conv.getWidth();
              inDepth = (int) conv.getDepth();
          } else {
              throw new IllegalStateException(
                      "Invalid input type: expected InputTypeConvolutional or InputTypeConvolutionalFlat."
                              + " Got: " + inputType);
          }
          return new int[]{inH, inW, inDepth};
      }
docs: Get heigh/width/channels as length 3 int[] from the InputType
  
  @param inputType Input type to get
  @return Length