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;
|
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 {
|
||||||
|
|
|
@ -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.
|
||||||
|
|
|
@ -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);
|
||||||
|
|
|
@ -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;
|
||||||
|
|
Loading…
Reference in New Issue
Block a user