add custom metadata in metadata_schema.

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

View File

@ -137,9 +137,6 @@ message GraphTrace {
TPU_TASK_INVOKE = 17;
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.
message CalculatorTrace {

View File

@ -112,10 +112,6 @@ struct TraceEvent {
static constexpr EventType GPU_TASK_INVOKE = GraphTrace::GPU_TASK_INVOKE;
static constexpr EventType TPU_TASK_INVOKE = GraphTrace::TPU_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.

View File

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

View File

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