docs: 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;
          }
      }