org.nd4j.autodiff.samediff.internal.DependencyTracker 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.autodiff.samediff.internal;

import lombok.NonNull;
import lombok.extern.slf4j.Slf4j;
import org.nd4j.base.Preconditions;
import org.nd4j.linalg.primitives.Pair;

import java.util.*;

/**
 * Dependenci tracker. See {@link AbstractDependencyTracker} for details
 *
 * @param  For a dependency X -> Y, Y has type T
 * @param  For a dependency X -> Y, X has type D
 */
@Slf4j
public class DependencyTracker extends AbstractDependencyTracker {

    @Override
    protected Map newTMap() {
        return new HashMap<>();
    }

    @Override
    protected Set newTSet() {
        return new HashSet<>();
    }

    @Override
    protected String toStringT(T t) {
        return t.toString();
    }

    @Override
    protected String toStringD(D d) {
        return d.toString();
    }
}