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