Examples with SameDiffOpExecutioner - org.nd4j.autodiff.samediff.SameDiffOpExecutioner

Example 1 with SameDiffOpExecutioner

use of org.nd4j.autodiff.samediff.SameDiffOpExecutioner in project nd4j by deeplearning4j.

the class SameDiffOpExecutionerTest method testupdateGraphFromProfiler.

@Test
public void testupdateGraphFromProfiler() {
    SameDiffOpExecutioner sameDiffOpExecutioner = new SameDiffOpExecutioner();
    Nd4j.getExecutioner().setProfilingMode(OpExecutioner.ProfilingMode.ALL);
    Nd4j.getExecutioner().exec(new Sigmoid(Nd4j.scalar(1.0)));
    SameDiff sameDiff = sameDiffOpExecutioner.getSameDiff();
}

Also used : SameDiff(org.nd4j.autodiff.samediff.SameDiff) SameDiffOpExecutioner(org.nd4j.autodiff.samediff.SameDiffOpExecutioner) Sigmoid(org.nd4j.linalg.api.ops.impl.transforms.Sigmoid) Test(org.junit.Test)

Aggregations

Test (org.junit.Test)1 SameDiff (org.nd4j.autodiff.samediff.SameDiff)1 SameDiffOpExecutioner (org.nd4j.autodiff.samediff.SameDiffOpExecutioner)1 Sigmoid (org.nd4j.linalg.api.ops.impl.transforms.Sigmoid)1