| Package | Description |
|---|---|
| org.nd4j.autodiff.listeners.profiler |
| Modifier and Type | Method and Description |
|---|---|
static ProfilingListener.Builder |
ProfilingListener.builder(File outputFile)
Create a new builder
|
ProfilingListener.Builder |
ProfilingListener.Builder.maxProfileIterations(int iterations)
Set a limit on the maximum number of iterations to profile (after warmup, if any).
|
ProfilingListener.Builder |
ProfilingListener.Builder.maxProfilerMilliseconds(long ms)
Set a limit on the maximum duration for profiling, in milliseconds.
|
ProfilingListener.Builder |
ProfilingListener.Builder.operations(Operation... operations)
Specify the operations (training, inference, etc) to profile.
|
ProfilingListener.Builder |
ProfilingListener.Builder.recordAll()
If called, all data will be profiled with no limits (other than a warmup, if set)
|
ProfilingListener.Builder |
ProfilingListener.Builder.warmup(int iterations)
Specify the number of warmup iterations - i.e., these will be excluded from profiling results
|
Copyright © 2021. All rights reserved.