Package org.nd4j.autodiff.samediff.transform

Interface SubGraphProcessor

  • public interface SubGraphProcessor

    • Method Summary

      All Methods Instance Methods Abstract Methods 
      Modifier and Type Method 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