package log_memory
- Alphabetic
- Public
- All
Type Members
-
final
case class
MemoryLogRawAllocation(stepId: Long = 0L, operation: String = "", numBytes: Long = 0L, ptr: Long = 0L, allocationId: Long = 0L, allocatorName: String = "", unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[MemoryLogRawAllocation] with Product with Serializable
- stepId
Process-unique step id.
- operation
Name of the operation making the allocation.
- numBytes
Number of bytes in the allocation.
- ptr
Address of the allocation.
- allocationId
Id of the tensor buffer being allocated, used to match to a corresponding deallocation.
- allocatorName
Name of the allocator used.
- Annotations
- @SerialVersionUID()
-
final
case class
MemoryLogRawDeallocation(stepId: Long = 0L, operation: String = "", allocationId: Long = 0L, allocatorName: String = "", deferred: Boolean = false, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[MemoryLogRawDeallocation] with Product with Serializable
- stepId
Process-unique step id.
- operation
Name of the operation making the deallocation.
- allocationId
Id of the tensor buffer being deallocated, used to match to a corresponding allocation.
- allocatorName
Name of the allocator used.
- deferred
True if the deallocation is queued and will be performed later, e.g. for GPU lazy freeing of buffers.
- Annotations
- @SerialVersionUID()
-
final
case class
MemoryLogStep(stepId: Long = 0L, handle: String = "", unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[MemoryLogStep] with Product with Serializable
- stepId
Process-unique step id.
- handle
Handle describing the feeds and fetches of the step.
- Annotations
- @SerialVersionUID()
-
final
case class
MemoryLogTensorAllocation(stepId: Long = 0L, kernelName: String = "", tensor: Option[TensorDescription] = _root_.scala.None, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[MemoryLogTensorAllocation] with Product with Serializable
- stepId
Process-unique step id.
- kernelName
Name of the kernel making the allocation as set in GraphDef, e.g., "affine2/weights/Assign".
- tensor
Allocated tensor details.
- Annotations
- @SerialVersionUID()
-
final
case class
MemoryLogTensorDeallocation(allocationId: Long = 0L, allocatorName: String = "", unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[MemoryLogTensorDeallocation] with Product with Serializable
- allocationId
Id of the tensor buffer being deallocated, used to match to a corresponding allocation.
- allocatorName
Name of the allocator used.
- Annotations
- @SerialVersionUID()
-
final
case class
MemoryLogTensorOutput(stepId: Long = 0L, kernelName: String = "", index: Int = 0, tensor: Option[TensorDescription] = _root_.scala.None, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[MemoryLogTensorOutput] with Product with Serializable
- stepId
Process-unique step id.
- kernelName
Name of the kernel producing an output as set in GraphDef, e.g., "affine2/weights/Assign".
- index
Index of the output being set.
- tensor
Output tensor details.
- Annotations
- @SerialVersionUID()
Value Members
- object LogMemoryProto extends GeneratedFileObject
- object MemoryLogRawAllocation extends GeneratedMessageCompanion[MemoryLogRawAllocation]
- object MemoryLogRawDeallocation extends GeneratedMessageCompanion[MemoryLogRawDeallocation]
- object MemoryLogStep extends GeneratedMessageCompanion[MemoryLogStep]
- object MemoryLogTensorAllocation extends GeneratedMessageCompanion[MemoryLogTensorAllocation]
- object MemoryLogTensorDeallocation extends GeneratedMessageCompanion[MemoryLogTensorDeallocation]
- object MemoryLogTensorOutput extends GeneratedMessageCompanion[MemoryLogTensorOutput]