Move common/constants and common/utils to xprof open source, and cons…

…olidate data table 1P/3P PiperOrigin-RevId: 713757127
tensorflow · Jan 9, 2025 · 5ec1558 · 5ec1558
1 parent d11ffb8
commit 5ec1558
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/frontend/app/common/interfaces/data_table.ts b/frontend/app/common/interfaces/data_table.ts
@@ -3,6 +3,7 @@ import * as memoryProfileProto from 'org_xprof/frontend/app/common/interfaces/me
 import * as memoryViewerPreprocess from 'org_xprof/frontend/app/common/interfaces/memory_viewer_preprocess.jsonpb_decls';
 import * as opProfileProto from 'org_xprof/frontend/app/common/interfaces/op_profile.jsonpb_decls';
 import * as diagnosticsProto from 'org_xprof/frontend/app/common/interfaces/diagnostics';
+import {RooflineModelData} from 'org_xprof/frontend/app/common/interfaces/roofline_model';
 
 /** Constant of empty data in SimpleDataTable typing */
 export const DEFAULT_SIMPLE_DATA_TABLE = {
@@ -423,3 +424,13 @@ export declare interface InferenceProfileData extends SimpleDataTable {
 /** All Inference Stats page data table type. */
 export type InferenceProfileTable =
     |InferenceProfileMetadata|InferenceProfileData;
+
+/* tslint:enable */
+
+/**
+ * The internal tool's response data type
+ * yet to be open sourced
+ */
+export type XprofDataTable =
+    |DataTable|InferenceProfileTable[]
+    |RooflineModelData[];
diff --git a/frontend/app/common/utils/BUILD b/frontend/app/common/utils/BUILD
@@ -19,6 +19,7 @@ ts_library(
     ],
     deps = [
         "@npm//@ngrx/store",
+        "@org_xprof//frontend/app/common/constants",
         "@org_xprof//frontend/app/common/interfaces",
         "@org_xprof//frontend/app/common/interfaces:chart",
         "@org_xprof//frontend/app/common/interfaces:op_profile_proto_defs",