diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-05 12:25:22 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-05 12:31:00 -0700 |
| commit | 58845f229be9b5ba2e1e36150bff5ba7a85920d8 (patch) | |
| tree | 34b42fb9c71aa2b4efa06d07d98dec38c0687374 /tensorflow/contrib | |
| parent | d016650ca7636c96c6664bed2cf3a2fa8a3c674b (diff) | |
Profiler collects the number of replicas and num cores per replica used in the model.
PiperOrigin-RevId: 215947354
Diffstat (limited to 'tensorflow/contrib')
| -rw-r--r-- | tensorflow/contrib/tpu/profiler/tf_op_stats.proto | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/contrib/tpu/profiler/tf_op_stats.proto b/tensorflow/contrib/tpu/profiler/tf_op_stats.proto index f88dc51636..1e66801efd 100644 --- a/tensorflow/contrib/tpu/profiler/tf_op_stats.proto +++ b/tensorflow/contrib/tpu/profiler/tf_op_stats.proto @@ -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. |