Interface OpImportFilter<GRAPH_TYPE,​NODE_TYPE,​ATTR_TYPE>

  • All Known Subinterfaces:
    TFOpImportFilter

    public interface OpImportFilter<GRAPH_TYPE,​NODE_TYPE,​ATTR_TYPE>
    Define whether the operation should be skipped during import
    • Method Detail

      • skipOp

        boolean skipOp​(NODE_TYPE nodeDef,
                       SameDiff initWith,
                       Map<String,​ATTR_TYPE> attributesForNode,
                       GRAPH_TYPE graph)
        If true: the op should be skipped for import, and its output variables should not be created. If false: the op should be imported
        Parameters:
        nodeDef - Node
        initWith - SameDiff instance
        attributesForNode - Attributes for the node
        graph - Graph to import from
        Returns:
        True if the op should be skipped during import