Uses of Class
org.nd4j.autodiff.samediff.transform.SubGraph
-
Packages that use SubGraph Package Description org.nd4j.autodiff.samediff.transform -
-
Uses of SubGraph in org.nd4j.autodiff.samediff.transform
Methods in org.nd4j.autodiff.samediff.transform that return SubGraph Modifier and Type Method Description SubGraphSubGraphPredicate. getSubGraph(SameDiff sd, DifferentialFunction rootFn)Get the SubGraph that matches the predicateMethods in org.nd4j.autodiff.samediff.transform that return types with arguments of type SubGraph Modifier and Type Method Description static List<SubGraph>GraphTransformUtil. getSubgraphsMatching(SameDiff sd, SubGraphPredicate p)Get a list of all the subgraphs that match the specified predicateMethods in org.nd4j.autodiff.samediff.transform with parameters of type SubGraph Modifier and Type Method Description List<SDVariable>SubGraphProcessor. processSubgraph(SameDiff sd, SubGraph subGraph)Replace the subgraph, and return the new outputs that should replace the old outputs.
Note that the order of the outputs you return matters!
If the original outputs are [A,B,C] and you return output variables [X,Y,Z], then anywhere "A" was used as input will now use "X"; similarly Y replaces B, and Z replaces C.
-