org.nd4j.imports.graphmapper.OpImportOverride Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of nd4j-api Show documentation

There is a newer version: 1.0.0-M2.1

package org.nd4j.imports.graphmapper;

import org.nd4j.autodiff.samediff.SDVariable;
import org.nd4j.autodiff.samediff.SameDiff;

import java.util.List;
import java.util.Map;

/**
 * An interface for overriding the import of an operation
 * @author Alex Black
 */
public interface OpImportOverride {

    /**
     * Initialize the operation and return its output variables
     */
    List initFromTensorFlow(List inputs, List controlDepInputs, NODE_TYPE nodeDef, SameDiff initWith, Map attributesForNode, GRAPH_TYPE graph);

}