org.nd4j.imports.graphmapper

Interface OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE>

All Known Subinterfaces:

TFImportOverride

public interface OpImportOverride<GRAPH_TYPE,NODE_TYPE,ATTR_TYPE>

An interface for overriding the import of an operation

Author:

Alex Black

Method Summary

Modifier and Type	Method and Description
List<SDVariable>	initFromTensorFlow(List<SDVariable> inputs, List<SDVariable> controlDepInputs, NODE_TYPE nodeDef, SameDiff initWith, Map<String,ATTR_TYPE> attributesForNode, GRAPH_TYPE graph) Initialize the operation and return its output variables

Method Detail

initFromTensorFlow

List<SDVariable> initFromTensorFlow(List<SDVariable> inputs,
                                    List<SDVariable> controlDepInputs,
                                    NODE_TYPE nodeDef,
                                    SameDiff initWith,
                                    Map<String,ATTR_TYPE> attributesForNode,
                                    GRAPH_TYPE graph)

Initialize the operation and return its output variables