cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add custom metadata in metadata_schema.

Browse Source 
PiperOrigin-RevId: 514768756
This commit is contained in:
Yuqi Li 2023-03-07 10:13:13 -08:00 committed by Copybara-Service
parent 414031867e
commit 46b5c4012d
4 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mediapipe/framework/calculator_profile.proto
View File

@ -137,9 +137,6 @@ message GraphTrace {
TPU_TASK_INVOKE = 17; TPU_TASK_INVOKE = 17;
CPU_TASK_INVOKE = 18; CPU_TASK_INVOKE = 18;
} }
// //depot/mediapipe/framework/mediapipe_profiling.h:profiler_census_tags,
// //depot/mediapipe/framework/profiler/trace_buffer.h:event_type_list,
// )
// The timing for one packet set being processed at one caclulator node. // The timing for one packet set being processed at one caclulator node.
message CalculatorTrace { message CalculatorTrace {

4
mediapipe/framework/profiler/trace_buffer.h
View File

@ -112,10 +112,6 @@ struct TraceEvent {
static constexpr EventType GPU_TASK_INVOKE = GraphTrace::GPU_TASK_INVOKE; static constexpr EventType GPU_TASK_INVOKE = GraphTrace::GPU_TASK_INVOKE;
static constexpr EventType TPU_TASK_INVOKE = GraphTrace::TPU_TASK_INVOKE; static constexpr EventType TPU_TASK_INVOKE = GraphTrace::TPU_TASK_INVOKE;
static constexpr EventType CPU_TASK_INVOKE = GraphTrace::CPU_TASK_INVOKE; static constexpr EventType CPU_TASK_INVOKE = GraphTrace::CPU_TASK_INVOKE;
// //depot/mediapipe/framework/mediapipe_profiling.h:profiler_census_tags,
// //depot/mediapipe/framework/calculator_profile.proto:event_type,
// )
}; };
// Packet trace log buffer. // Packet trace log buffer.

2
mediapipe/tasks/cc/metadata/metadata_version.cc
View File

@ -285,7 +285,7 @@ void UpdateMinimumVersionForTable<tflite::SubGraphMetadata>(
min_version); min_version);
} }
// Checks for the options field. // Checks for the custom metadata field.
if (table->custom_metadata() != nullptr) { if (table->custom_metadata() != nullptr) {
UpdateMinimumVersion(GetMemberVersion(SchemaMembers::kCustomMetadata), UpdateMinimumVersion(GetMemberVersion(SchemaMembers::kCustomMetadata),
min_version); min_version);

3
mediapipe/tasks/metadata/metadata_schema.fbs
View File

@ -680,6 +680,8 @@ table SubGraphMetadata {
// Added in: 1.2.0 // Added in: 1.2.0
output_tensor_groups:[TensorGroup]; output_tensor_groups:[TensorGroup];
// A list of custom metadata.
// Added in: 1.5.0.
custom_metadata:[CustomMetadata]; custom_metadata:[CustomMetadata];
} }
@ -726,6 +728,5 @@ table ModelMetadata {
// the metadata is populated into a TFLite model. // the metadata is populated into a TFLite model.
min_parser_version:string; min_parser_version:string;
} }
// metadata_version.cc)
root_type ModelMetadata; root_type ModelMetadata;