org.nd4j.imports.graphmapper

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 Summary

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

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