add custom metadata in metadata_schema.
PiperOrigin-RevId: 514768756
This commit is contained in:
parent
414031867e
commit
46b5c4012d
|
@ -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 {
|
||||
|
|
|
@ -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.
|
||||
|
|
|
@ -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);
|
||||
|
|
|
@ -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;
|
||||
|
|
Loading…
Reference in New Issue
Block a user