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

Profiler collects the number of replicas and num cores per replica used in the model.

Browse Source 
PiperOrigin-RevId: 215947354
This commit is contained in:
A. Unique TensorFlower 2018-10-05 12:25:22 -07:00 committed by TensorFlower Gardener
parent d016650ca7
commit 58845f229b
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/contrib/tpu/profiler/tf_op_stats.proto
View File

@ -168,6 +168,12 @@ message RunEnvironmentResult {
optional HostIndependentJobInfoResult host_independent_job_info = 5;
// Host-dependent job information.
repeated HostDependentJobInfoResult host_dependent_job_info = 6;
// The number of replicas, corresponds to input parallelism.
// If there is no model parallelism, replica_count = tpu_core_count
optional int32 replica_count = 7;
// The number of cores used for a single replica, e.g. model parallelism.
// If there is no model parallelism, then num_cores_per_replica = 1
optional int32 num_cores_per_replica = 8;
}
// The types of host operations that are tracked.