package step_stats
- Alphabetic
- Public
- All
Type Members
-
final
case class
AllocationRecord(allocMicros: Long = 0L, allocBytes: Long = 0L, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[AllocationRecord] with Product with Serializable
An allocation/de-allocation operation performed by the allocator.
An allocation/de-allocation operation performed by the allocator.
- allocMicros
The timestamp of the operation.
- allocBytes
Number of bytes allocated, or de-allocated if negative.
- Annotations
- @SerialVersionUID()
-
final
case class
AllocatorMemoryUsed(allocatorName: String = "", totalBytes: Long = 0L, peakBytes: Long = 0L, liveBytes: Long = 0L, allocationRecords: Seq[AllocationRecord] = _root_.scala.Seq.empty, allocatorBytesInUse: Long = 0L, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[AllocatorMemoryUsed] with Product with Serializable
- totalBytes
These are per-node allocator memory stats.
- liveBytes
The bytes that are not deallocated.
- allocationRecords
The allocation and deallocation timeline.
- allocatorBytesInUse
These are snapshots of the overall allocator memory stats. The number of live bytes currently allocated by the allocator.
- Annotations
- @SerialVersionUID()
-
final
case class
DeviceStepStats(device: String = "", nodeStats: Seq[NodeExecStats] = _root_.scala.Seq.empty, threadNames: Map[Int, String] = ..., unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[DeviceStepStats] with Product with Serializable
- threadNames
Its key is thread id.
- Annotations
- @SerialVersionUID()
-
final
case class
MemoryStats(tempMemorySize: Long = 0L, persistentMemorySize: Long = 0L, persistentTensorAllocIds: Seq[Long] = _root_.scala.Seq.empty, deviceTempMemorySize: Long = 0L, devicePersistentMemorySize: Long = 0L, devicePersistentTensorAllocIds: Seq[Long] = _root_.scala.Seq.empty, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[MemoryStats] with Product with Serializable
For memory tracking.
For memory tracking.
- Annotations
- @SerialVersionUID()
-
final
case class
NodeExecStats(nodeName: String = "", allStartMicros: Long = 0L, opStartRelMicros: Long = 0L, opEndRelMicros: Long = 0L, allEndRelMicros: Long = 0L, memory: Seq[AllocatorMemoryUsed] = _root_.scala.Seq.empty, output: Seq[NodeOutput] = _root_.scala.Seq.empty, timelineLabel: String = "", scheduledMicros: Long = 0L, threadId: Int = 0, referencedTensor: Seq[AllocationDescription] = _root_.scala.Seq.empty, memoryStats: Option[MemoryStats] = _root_.scala.None, allStartNanos: Long = 0L, opStartRelNanos: Long = 0L, opEndRelNanos: Long = 0L, allEndRelNanos: Long = 0L, scheduledNanos: Long = 0L, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[NodeExecStats] with Product with Serializable
Time/size stats recorded for a single execution of a graph node.
Time/size stats recorded for a single execution of a graph node.
- nodeName
TODO(tucker): Use some more compact form of node identity than the full string name. Either all processes should agree on a global id (cost_id?) for each node, or we should use a hash of the name.
- Annotations
- @SerialVersionUID()
-
final
case class
NodeOutput(slot: Int = 0, tensorDescription: Option[TensorDescription] = _root_.scala.None, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[NodeOutput] with Product with Serializable
Output sizes recorded for a single execution of a graph node.
Output sizes recorded for a single execution of a graph node.
- Annotations
- @SerialVersionUID()
-
final
case class
StepStats(devStats: Seq[DeviceStepStats] = _root_.scala.Seq.empty, unknownFields: UnknownFieldSet = ...) extends GeneratedMessage with Updatable[StepStats] with Product with Serializable
- Annotations
- @SerialVersionUID()
Value Members
- object AllocationRecord extends GeneratedMessageCompanion[AllocationRecord]
- object AllocatorMemoryUsed extends GeneratedMessageCompanion[AllocatorMemoryUsed]
- object DeviceStepStats extends GeneratedMessageCompanion[DeviceStepStats]
- object MemoryStats extends GeneratedMessageCompanion[MemoryStats]
- object NodeExecStats extends GeneratedMessageCompanion[NodeExecStats]
- object NodeOutput extends GeneratedMessageCompanion[NodeOutput]
- object StepStats extends GeneratedMessageCompanion[StepStats]
- object StepStatsProto extends GeneratedFileObject