aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Benoit Steiner <bsteiner@google.com>2017-08-29 11:31:51 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-08-29 11:35:59 -0700
commitccc81808f6f5e2b1f593ea668ec1d3d89710da8c (patch)
treeecad39259f2ca7846cf83da8763837da99f4ade3
parenta7aa47fd9f23e91ab84eb0f75ff6a8d5b6b95930 (diff)
Annotate the performance database with expected distribution of values.
PiperOrigin-RevId: 166876854
-rw-r--r--tensorflow/core/grappler/costs/op_performance_data.proto16
1 files changed, 16 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/op_performance_data.proto b/tensorflow/core/grappler/costs/op_performance_data.proto
index 0d6b337d5a..1a111b71dc 100644
--- a/tensorflow/core/grappler/costs/op_performance_data.proto
+++ b/tensorflow/core/grappler/costs/op_performance_data.proto
@@ -48,6 +48,16 @@ message OpInfo {
DeviceProperties device = 4;
}
+message NormalDistribution {
+ double mu = 1;
+ double sigma = 2;
+}
+
+message LogNormalDistribution {
+ double mu = 1;
+ double sigma = 2;
+}
+
// Performance data for tensorflow operations
message OpPerformance {
// The op
@@ -75,6 +85,12 @@ message OpPerformance {
// Percentage of theoretical memory performance.
double memory_efficiency = 8;
+ // Expected execution time, modeled using one of 2 possible distributions.
+ oneof execution_time {
+ NormalDistribution execution_time_normal = 10;
+ LogNormalDistribution execution_time_log_normal = 11;
+ };
+
// Memory usage data for a tensorflow operation.
message OpMemory {
// The output information may have memory usage and output shapes.