public static DataType getDtypeFromContext ( String dType ) { switch ( dType ) { case "double" : return DataType . DOUBLE ; case "float" : return DataType . FLOAT ; case "int" : return DataType . INT ; case "half" : return DataType . HALF ; default : return DataType . FLOAT ; } }
Get the allocation mode from the context