@if (loading()) {
}
@if (sidebarExpanded()) { Memory Breakdown }
@if (sidebarExpanded()) { @if (result && spaceList.length > 0) {
Memory Space
@if (spaceList.length > 1) {
All Spaces
{{totalMemoryMib | number:'1.1-2'}} MiB
} @for (space of result.memorySpaceBreakdown | keyvalue:compareSpacesDesc; track space.key) {
{{space.key}}
{{space.value | number:'1.1-2'}} MiB
}
} @if (activeCategorySummaries().length > 0) {
Categories
@for (summary of activeCategorySummaries(); track summary.name) {
{{summary.name}}
{{summary.percentage | number:'1.1-1'}}%
{{summary.value | number:'1.1-2'}} MiB
}
} }
@if (moduleList.length > 1) {
Select HLO Module @for (mod of moduleList; track mod) { {{mod}} }
}
JAX Tree Category Tree Total Size Padding Size Category Color Padding Color Classic Color
@if (rootNode) { } @if (rootNode) { }
Search buffers, ops, or paths search @if (dtypeList.length > 1) { Dtype @for (dt of dtypeList; track dt) { {{dt}} } }
Buffer Name {{row.name}} Space {{row.memorySpace}} Size (MiB) {{row.sizeMib | number:'1.1-2'}} Category {{row.category}} Sub-category {{row.subCategory}} Group {{row.group}} Padding (MiB) {{row.paddingMib | number:'1.1-2'}} ({{row.paddingPercentage | number:'1.1-1'}}%) Dtype {{row.dtype}} Shape {{row.shape}} TF Op {{row.tfOpName}} JAX Path {{row.jaxVariablePath}}