org.nd4j.autodiff.samediff.transform

Interface SubGraphProcessor

public interface SubGraphProcessor

Method Summary

Modifier and Type	Method and Description
List<SDVariable>	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.

Method Detail

processSubgraph

List<SDVariable> 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.

Parameters:

sd - SameDiff instance

subGraph - Subgraph to modify

Returns:

New output variables