code: public DataType nd4jTypeFromOnnxType(OnnxProto3.TensorProto.DataType dataType) {
          switch (dataType) {
              case DOUBLE: return DataType.DOUBLE;
              case FLOAT: return DataType.FLOAT;
              case FLOAT16: return DataType.HALF;
              case INT32:
              case INT64: return DataType.INT;
              default: return DataType.UNKNOWN;
          }
      }
docs: Convert an onnx type to the proper nd4j type
  @param dataType the data type to convert
  @return the nd4j type for the onnx type