org.nd4j.linalg.api.parallel.ParallelExecutioner Maven / Gradle / Ivy

Go to download

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.linalg.api.parallel;

import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.api.ops.Accumulation;
import org.nd4j.linalg.api.ops.Op;
import org.nd4j.linalg.api.ops.executioner.OpExecutioner;

import java.util.concurrent.ForkJoinTask;
import java.util.concurrent.Future;

/**
 * Executes operations in parallel
 *
 * @author Adam Gibson
 */
public interface ParallelExecutioner {


    /**
     * Sets parallel enabled
     * @param parallelEnabled
     */
    void setParallelEnabled(boolean parallelEnabled);
    /**
     * Whether the parallel execution is enabled
     * @return true if the parallel execution is enabled
     * false otherwise
     */
    boolean parallelEnabled();

    /**
     *
     * @param arr
     * @param task
     * @param dimension
     */
    INDArray execBasedOnArraysAlongDimension(INDArray arr, Accumulation task, OpExecutioner executioner, int... dimension);


    /**
     *
     * @param arr
     * @param task
     * @param dimension
     */
    void execBasedOnArraysAlongDimension(INDArray arr, Op task, OpExecutioner executioner, int... dimension);

    /**
     *
     * @param arr
     * @param task
     */
    void execBasedOnSlices(INDArray arr,Op task,OpExecutioner executioner);

    /**
     *
     * @param arr
     * @param task
     * @param dimension
     */
    void execBasedOnArraysAlongDimension(INDArray arr, TaskCreator.INDArrayTask task, int... dimension);


    /**
     *
     * @param arr
     * @param task
     * @param dimension
     */
    void execBasedOnArraysAlongDimension(INDArray[] arr, TaskCreator.INDArrayTask task, int... dimension);

    /**
     *
     * @param arr
     * @param task
     */
    void execBasedOnSlices(INDArray arr,TaskCreator.INDArrayTask task);

    /**
     *
     * @param runnable
     * @return
     */
    Future exec(Runnable runnable);

    /**
     *
     * @param task
     * @param 
     */
     void  exec(ForkJoinTask task);

}