@Override public NodeDef getNodeWithNameFromGraph(GraphDef graph, String name) { for(int i = 0; i < graph.getNodeCount(); i++) { val node = graph.getNode(i); if(node.getName().equals(name)) return node; } return null; }
@Override public void initFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String, AttrValue> attributesForNode, GraphDef graph) { val idd = nodeDef.getInput(nodeDef.getInputCount() - 1); NodeDef iddNode = null; for(int i = 0; i < graph.getNodeCount(); i++) { if(graph.getNode(i).getName().equals(idd)) { iddNode = graph.getNode(i); } } val arr = TFGraphMapper.getInstance().getNDArrayFromTensor("value",iddNode,graph); if (arr != null) { int idx = arr.getInt(0); addIArgument(idx); } }
@java.lang.Override public int hashCode() { if (memoizedHashCode != 0) { return memoizedHashCode; } int hash = 41; hash = (19 * hash) + getDescriptor().hashCode(); if (getNodeCount() > 0) { hash = (37 * hash) + NODE_FIELD_NUMBER; hash = (53 * hash) + getNodeList().hashCode(); } if (hasVersions()) { hash = (37 * hash) + VERSIONS_FIELD_NUMBER; hash = (53 * hash) + getVersions().hashCode(); } hash = (37 * hash) + VERSION_FIELD_NUMBER; hash = (53 * hash) + getVersion(); if (hasLibrary()) { hash = (37 * hash) + LIBRARY_FIELD_NUMBER; hash = (53 * hash) + getLibrary().hashCode(); } hash = (29 * hash) + unknownFields.hashCode(); memoizedHashCode = hash; return hash; }
for(int i = 0; i < graph.getNodeCount(); i++) { if (graph.getNode(i).getName().equals(nodeDef.getName() + "/reduction_indices")) { reductionNode = graph.getNode(i);
val shapeNode = nodeDef.getInput(1); NodeDef shapeNodeInGraph = null; for (int i = 0; i < graph.getNodeCount(); i++) { if (graph.getNode(i).getName().equals(shapeNode)) { shapeNodeInGraph = graph.getNode(i);
NodeDef endNode = null; for(int i = 0; i < graph.getNodeCount(); i++) { if(graph.getNode(i).getName().equals(inputBegin)) { beginNode = graph.getNode(i);
NodeDef strides = null; for(int i = 0; i < graph.getNodeCount(); i++) { if(graph.getNode(i).getName().equals(inputBegin)) { beginNode = graph.getNode(i);
return; NodeDef permuteDimsNode = null; for (int i = 0; i < graph.getNodeCount(); i++) { if (graph.getNode(i).getName().equals(nodeDef.getInput(1))) { permuteDimsNode = graph.getNode(i);
private void initSessionAndStatusIfNeeded(org.tensorflow.framework.GraphDef graphDef1) { for(int i = 0; i < graphDef1.getNodeCount(); i++) { NodeDef node = graphDef1.getNode(i); for(int input = 0; input < node.getInputCount(); input++) { log.trace("Attempting to automatically resolve tensorflow output names.."); for(int i = 0; i < graphDef1.getNodeCount(); i++) { if(!seenAsInput.contains(graphDef1.getNode(i).getName()) && !graphDef1.getNode(i).getOp().equals("Placeholder")) { outputOrder.add(graphDef1.getNode(i).getName());
@java.lang.Override public int hashCode() { if (memoizedHashCode != 0) { return memoizedHashCode; } int hash = 41; hash = (19 * hash) + getDescriptor().hashCode(); if (getNodeCount() > 0) { hash = (37 * hash) + NODE_FIELD_NUMBER; hash = (53 * hash) + getNodeList().hashCode(); } if (hasVersions()) { hash = (37 * hash) + VERSIONS_FIELD_NUMBER; hash = (53 * hash) + getVersions().hashCode(); } hash = (37 * hash) + VERSION_FIELD_NUMBER; hash = (53 * hash) + getVersion(); if (hasLibrary()) { hash = (37 * hash) + LIBRARY_FIELD_NUMBER; hash = (53 * hash) + getLibrary().hashCode(); } hash = (29 * hash) + unknownFields.hashCode(); memoizedHashCode = hash; return hash; }