diff --git a/frontend/app/common/interfaces/BUILD b/frontend/app/common/interfaces/BUILD index 426319362..d748e5bd7 100644 --- a/frontend/app/common/interfaces/BUILD +++ b/frontend/app/common/interfaces/BUILD @@ -32,16 +32,6 @@ genrule( ], ) -genrule( - name = "memory_viewer_preprocess_proto_ts", - srcs = ["memory_viewer_preprocess.jsonpb_decls.d.ts.gz"], - outs = ["memory_viewer_preprocess.jsonpb_decls.d.ts"], - cmd = "gzip -d -c $(SRCS) > $(OUTS)", - visibility = [ - "//frontend:__subpackages__", "//plugin:__subpackages__", - ], -) - ts_library( name = "op_metrics_proto_defs", srcs = [ diff --git a/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts b/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts new file mode 100644 index 000000000..75d5cb670 --- /dev/null +++ b/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts @@ -0,0 +1,63 @@ +// This file must be kept in sync with the corresponding proto file. + +/** See corresponding proto */ +export interface HeapObject { + numbered?: /* int32 */ number; + named?: string; + label?: string; + logicalBufferId?: /* int32 */ number; + logicalBufferSizeMib?: /* double */ number; + unpaddedShapeMib?: /* double */ number; + instructionName?: string; + shapeString?: string; + tfOpName?: string; + groupName?: string; + opCode?: string; +} + +/** See corresponding proto */ +export interface BufferSpan { + start?: /* int32 */ number; + limit?: /* int32 */ number; +} + +/** See corresponding proto */ +export interface LogicalBuffer { + id?: /* int64 */ string; + shape?: string; + sizeMib?: /* double */ number; + hloName?: string; + shapeIndex?: /* int64 */ string[]; +} + +/** See corresponding proto */ +export interface BufferAllocation { + id?: /* int64 */ string; + sizeMib?: /* double */ number; + attributes?: string[]; + logicalBuffers?: LogicalBuffer[]; + commonShape?: string; +} + +/** See corresponding proto */ +export interface PreprocessResult { + heapSizes?: /* double */ number[]; + unpaddedHeapSizes?: /* double */ number[]; + maxHeap?: HeapObject[]; + maxHeapBySize?: HeapObject[]; + logicalBufferSpans?: {[key: /* int32 */ string]: BufferSpan}; + maxHeapToBySize?: /* int32 */ number[]; + bySizeToMaxHeap?: /* int32 */ number[]; + moduleName?: string; + entryComputationName?: string; + peakHeapMib?: /* double */ number; + peakUnpaddedHeapMib?: /* double */ number; + peakHeapSizePosition?: /* int32 */ number; + entryComputationParametersMib?: /* double */ number; + nonReusableMib?: /* double */ number; + maybeLiveOutMib?: /* double */ number; + totalBufferAllocationMib?: /* double */ number; + indefiniteBufferAllocationMib?: /* double */ number; + indefiniteLifetimes?: BufferAllocation[]; + allocationTimeline?: string; +} diff --git a/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts.gz b/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts.gz deleted file mode 100644 index 610feddfa..000000000 Binary files a/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls.d.ts.gz and /dev/null differ