org.nd4j.autodiff.opstate.OpExecOrder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of samediff

There is a newer version: 0.9.1

package org.nd4j.autodiff.opstate;

import lombok.Builder;
import lombok.Data;

import java.util.ArrayList;
import java.util.List;

/**
 * Created by agibsonccc on 5/1/17.
 */
@Data
@Builder
public class OpExecOrder {
    private List actions;


    /**
     *
     * @return
     */
    public List opStates() {
        List ret = new ArrayList<>(actions.size());
        for(OpExecAction opExecAction : actions) {
            ret.add(opExecAction.getOpState());
        }

        return ret;
    }

    public List opNames() {
        List ret = new ArrayList<>(actions.size());
        for(OpExecAction opExecOrder : actions) {
            ret.add(opExecOrder.getOpState().getOpName());
        }

        return ret;
    }





}