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 ; } }
Convert an onnx type to the proper nd4j type