Class ProfilingListener.Builder
- java.lang.Object
-
- org.nd4j.autodiff.listeners.profiler.ProfilingListener.Builder
-
- Enclosing class:
- ProfilingListener
public static class ProfilingListener.Builder extends Object
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ProfilingListener
build()
Create the profiling listenerProfilingListener.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
-
-
-
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
-
-