org.nd4j.autodiff.listeners.profiler

Class ProfilingListener.Builder

java.lang.Object

org.nd4j.autodiff.listeners.profiler.ProfilingListener.Builder

Enclosing class:

ProfilingListener

public static class ProfilingListener.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder(@NonNull File outputFile)

Method Summary

Modifier and Type	Method and Description
ProfilingListener	build() Create the profiling listener
ProfilingListener.Builder	maxProfileIterations(int iterations) Set a limit on the maximum number of iterations to profile (after warmup, if any).
ProfilingListener.Builder	maxProfilerMilliseconds(long ms) Set a limit on the maximum duration for profiling, in milliseconds.
ProfilingListener.Builder	operations(Operation... operations) Specify the operations (training, inference, etc) to profile.
ProfilingListener.Builder	recordAll() If called, all data will be profiled with no limits (other than a warmup, if set)
ProfilingListener.Builder	warmup(int iterations) Specify the number of warmup iterations - i.e., these will be excluded from profiling results

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Builder

public Builder(@NonNull
               @NonNull File outputFile)

Method Detail

recordAll

public ProfilingListener.Builder recordAll()

If called, all data will be profiled with no limits (other than a warmup, if set)

warmup

public ProfilingListener.Builder warmup(int iterations)

Specify the number of warmup iterations - i.e., these will be excluded from profiling results

maxProfileIterations

public ProfilingListener.Builder maxProfileIterations(int iterations)

Set a limit on the maximum number of iterations to profile (after warmup, if any). Any ops executed after the specified number of iterations will not be profiled/recorded

maxProfilerMilliseconds

public ProfilingListener.Builder maxProfilerMilliseconds(long ms)

Set a limit on the maximum duration for profiling, in milliseconds. Any ops executed after the specified amount of time since the first (non-warmup) operation start will not be profiled/recorded

operations

public ProfilingListener.Builder operations(Operation... operations)

Specify the operations (training, inference, etc) to profile. If not set, all operations are profiled

build

public ProfilingListener build()

Create the profiling listener