Add IOS_BENCHMARK enum value to BenchmarkType

PiperOrigin-RevId: 311481101
Change-Id: I142b5b6231a817df6b688786fa508379ce06dd79
This commit is contained in:
YoungSeok Yoon 2020-05-14 00:11:31 -07:00 committed by TensorFlower Gardener
parent d33cb73389
commit d36ad412f4

View File

@ -1,6 +1,8 @@
// Protocol messages for describing the results of benchmarks and unit tests. // Protocol messages for describing the results of benchmarks and unit tests.
syntax = "proto3"; syntax = "proto3";
package tensorflow;
import "google/protobuf/any.proto"; import "google/protobuf/any.proto";
import "google/protobuf/wrappers.proto"; import "google/protobuf/wrappers.proto";
@ -9,14 +11,12 @@ option java_outer_classname = "TestLogProtos";
option java_multiple_files = true; option java_multiple_files = true;
option java_package = "org.tensorflow.util.testlog"; option java_package = "org.tensorflow.util.testlog";
package tensorflow;
message EntryValue { message EntryValue {
oneof kind { oneof kind {
double double_value = 1; double double_value = 1;
string string_value = 2; string string_value = 2;
} }
}; }
message MetricEntry { message MetricEntry {
// Metric name // Metric name
@ -62,7 +62,7 @@ message BenchmarkEntry {
// Metric name, value and expected range. This can include accuracy metrics // Metric name, value and expected range. This can include accuracy metrics
// typically used to determine whether the accuracy test has passed // typically used to determine whether the accuracy test has passed
repeated MetricEntry metrics = 7; repeated MetricEntry metrics = 7;
}; }
message BenchmarkEntries { message BenchmarkEntries {
repeated BenchmarkEntry entry = 1; repeated BenchmarkEntry entry = 1;
@ -72,7 +72,7 @@ message BuildConfiguration {
string mode = 1; // opt, dbg, etc string mode = 1; // opt, dbg, etc
repeated string cc_flags = 2; // CC compiler flags, if known repeated string cc_flags = 2; // CC compiler flags, if known
repeated string opts = 3; // Bazel compilation options, if known repeated string opts = 3; // Bazel compilation options, if known
}; }
message CommitId { message CommitId {
oneof kind { oneof kind {
@ -85,7 +85,7 @@ message CommitId {
string snapshot = 3; string snapshot = 3;
// Changelist tested if the change list is not already submitted. // Changelist tested if the change list is not already submitted.
int64 pending_changelist = 4; int64 pending_changelist = 4;
}; }
message CPUInfo { message CPUInfo {
int64 num_cores = 1; int64 num_cores = 1;
@ -105,7 +105,7 @@ message CPUInfo {
// Cache sizes (in bytes), e.g. "L2": 262144 (for 256KB) // Cache sizes (in bytes), e.g. "L2": 262144 (for 256KB)
map<string, int64> cache_size = 6; map<string, int64> cache_size = 6;
}; }
message MemoryInfo { message MemoryInfo {
int64 total = 1; // Total virtual memory in bytes int64 total = 1; // Total virtual memory in bytes
@ -116,7 +116,7 @@ message GPUInfo {
string model = 1; // e.g. "Tesla K40c" string model = 1; // e.g. "Tesla K40c"
string uuid = 2; // Final entry in output of "nvidia-smi -L" string uuid = 2; // Final entry in output of "nvidia-smi -L"
string bus_id = 3; // e.g. "0000:04:00.0" string bus_id = 3; // e.g. "0000:04:00.0"
}; }
message PlatformInfo { message PlatformInfo {
string bits = 1; // e.g. '64bit' string bits = 1; // e.g. '64bit'
@ -125,14 +125,14 @@ message PlatformInfo {
string release = 4; // e.g. '3.13.0-76-generic' string release = 4; // e.g. '3.13.0-76-generic'
string system = 5; // e.g. 'Linux' string system = 5; // e.g. 'Linux'
string version = 6; // e.g. '#120-Ubuntu SMP Mon Jan 18 15:59:10 UTC 2016' string version = 6; // e.g. '#120-Ubuntu SMP Mon Jan 18 15:59:10 UTC 2016'
}; }
message AvailableDeviceInfo { // Matches DeviceAttributes message AvailableDeviceInfo { // Matches DeviceAttributes
string name = 1; // Device name. string name = 1; // Device name.
string type = 2; // Device type, e.g. 'CPU' or 'GPU'. string type = 2; // Device type, e.g. 'CPU' or 'GPU'.
int64 memory_limit = 3; // Memory capacity in bytes. int64 memory_limit = 3; // Memory capacity in bytes.
string physical_description = 4; // The physical description of this device. string physical_description = 4; // The physical description of this device.
}; }
message MachineConfiguration { message MachineConfiguration {
// Host name of machine that ran the benchmark. // Host name of machine that ran the benchmark.
@ -154,7 +154,7 @@ message MachineConfiguration {
repeated AvailableDeviceInfo available_device_info = 5; repeated AvailableDeviceInfo available_device_info = 5;
MemoryInfo memory_info = 6; MemoryInfo memory_info = 6;
}; }
// Run-specific items such as arguments to the test / benchmark. // Run-specific items such as arguments to the test / benchmark.
message RunConfiguration { message RunConfiguration {
@ -206,6 +206,7 @@ message TestResults {
PYTHON_BENCHMARK = 2; PYTHON_BENCHMARK = 2;
ANDROID_BENCHMARK = 3; ANDROID_BENCHMARK = 3;
EDGE_BENCHMARK = 4; EDGE_BENCHMARK = 4;
IOS_BENCHMARK = 5;
} }
BenchmarkType benchmark_type = 10; BenchmarkType benchmark_type = 10;
@ -219,4 +220,4 @@ message TestResults {
// TensorFlow version this benchmark runs against. // TensorFlow version this benchmark runs against.
// This can be either set to full version or just the major version. // This can be either set to full version or just the major version.
string tf_version = 12; string tf_version = 12;
}; }