Examples with OpDef org.tensorflow.framework.OpDef used on opensource projects

Example 1 with OpDef

use of org.tensorflow.framework.OpDef in project nd4j by deeplearning4j.

the class TensorflowDescriptorParser method opDescs.

/**
 * Get the op descriptors for tensorflow
 * @return the op descriptors for tensorflow
 * @throws Exception
 */
public static Map<String, OpDef> opDescs() throws Exception {
    InputStream contents = new ClassPathResource("ops.proto").getInputStream();
    try (BufferedInputStream bis2 = new BufferedInputStream(contents);
        BufferedReader reader = new BufferedReader(new InputStreamReader(bis2))) {
        org.tensorflow.framework.OpList.Builder builder = org.tensorflow.framework.OpList.newBuilder();
        StringBuilder str = new StringBuilder();
        String line = null;
        while ((line = reader.readLine()) != null) {
            // .append("\n");
            str.append(line);
        }
        TextFormat.getParser().merge(str.toString(), builder);
        List<OpDef> list = builder.getOpList();
        Map<String, OpDef> map = new HashMap<>();
        for (OpDef opDef : list) {
            map.put(opDef.getName(), opDef);
        }
        return map;
    } catch (Exception e2) {
        e2.printStackTrace();
    }
    throw new ND4JIllegalStateException("Unable to load tensorflow descriptors!");
}