Package 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
All Methods Instance Methods Abstract Methods Modifier and Type Method Description booleanskipOp(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- NodeinitWith- SameDiff instanceattributesForNode- Attributes for the nodegraph- Graph to import from- Returns:
- True if the op should be skipped during import
-
-