Class | Description |
---|---|
ActivationDescriptor | |
AllocatorState | |
BlockInfo | |
CheckpointDelta | |
Constant | |
ConvolutionDescriptor | |
CTCLossDescriptor | |
CUDAAllocator | |
CUDAGuard |
A variant of DeviceGuard that is specialized for CUDA.
|
CUDAKernelLaunchInfo |
Use to hold info about kernel launches so that we can run kernels
asynchronously and still associate launches with device-side
assertion failures
|
CUDAKernelLaunchInfoVector | |
CUDAKernelLaunchInfoVector.Iterator | |
CUDAKernelLaunchRegistry |
Circular buffer used to hold information about kernel launches
this is later used to reconstruct how a device-side kernel assertion failure
occurred CUDAKernelLaunchRegistry is used as a singleton
|
CUDAMultiStreamGuard |
A variant of MultiStreamGuard that is specialized for CUDA.
|
CUDAStream | |
CUDAStreamArrayRef |
ArrayRef - Represent a constant reference to an array (0 or more elements
consecutively in memory), i.e.
|
CUDAStreamCaptureModeGuard | |
CUDAStreamGuard |
A variant of StreamGuard that is specialized for CUDA.
|
CuDNNError | |
DeviceAssertionData |
Holds information about any device-side assertions that fail.
|
DeviceAssertionsData |
Used to hold assertions generated by the device
Held in managed memory and access by both the CPU and the GPU.
|
DeviceAssertionsDataVector | |
DeviceAssertionsDataVector.Iterator | |
DeviceAssertionsDataVectorCUDAKernelLaunchInfoVectorPair | |
DeviceStats | |
DropoutDescriptor | |
FilterDescriptor | |
PointerSet | |
PointerSet.Iterator | |
RNNDescriptor | |
SegmentInfo | |
SnapshotInfo | |
SpatialTransformerDescriptor | |
Stat | |
TensorDescriptor | |
TraceEntry | |
TraceEntryVector | |
TraceEntryVector.Iterator |
Enum | Description |
---|---|
CUDAStream.Unchecked | |
TraceEntry.Action |
Copyright © 2024. All rights reserved.