experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Move session info into op info.

Browse Source 
PiperOrigin-RevId: 188912426
This commit is contained in:
Yuefeng Zhou 2018-03-13 12:29:55 -07:00 committed by TensorFlower Gardener
parent e4b0ddbeae
commit 571ed3ba4f
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/core/grappler/costs/op_performance_data.proto
View File

@ -24,6 +24,11 @@ import "tensorflow/core/framework/types.proto";
import "tensorflow/core/framework/attr_value.proto";
import "tensorflow/core/protobuf/device_properties.proto";
// Description of the session when an op is run.
message SessionInfo {
int64 intra_op_parallelism = 1;
}
// Description of an operation as well as the parameters expected to impact its
// performance.
message OpInfo {
@ -46,6 +51,9 @@ message OpInfo {
// Device on which the operation is run.
DeviceProperties device = 4;
// Information about the session configs.
SessionInfo session_info = 6;
}
message NormalDistribution {
@ -58,17 +66,13 @@ message LogNormalDistribution {
double sigma = 2;
}
message SessionInfo {
int64 intra_op_parallelism = 1;
}
// Performance data for tensorflow operations
message OpPerformance {
// The op
OpInfo op = 1;
// Information about the session configs.
SessionInfo session_info = 12;
SessionInfo session_info = 12 [deprecated = true];
// The node name (optional). Makes it easier to associate the performance data
// with a specific graph node.