Add IOS_BENCHMARK enum value to BenchmarkType
PiperOrigin-RevId: 311481101 Change-Id: I142b5b6231a817df6b688786fa508379ce06dd79
This commit is contained in:
parent
d33cb73389
commit
d36ad412f4
@ -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
|
||||
@ -113,26 +113,26 @@ message MemoryInfo {
|
||||
}
|
||||
|
||||
message GPUInfo {
|
||||
string model = 1; // e.g. "Tesla K40c"
|
||||
string uuid = 2; // Final entry in output of "nvidia-smi -L"
|
||||
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'
|
||||
string linkage = 2; // e.g. 'ELF'
|
||||
string machine = 3; // e.g. 'i386'
|
||||
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'
|
||||
};
|
||||
string bits = 1; // e.g. '64bit'
|
||||
string linkage = 2; // e.g. 'ELF'
|
||||
string machine = 3; // e.g. 'i386'
|
||||
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;
|
||||
};
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user