diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-18 14:28:45 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-18 14:35:44 -0800 |
| commit | 900f84b82d0658f086d40c0c026697898bf62132 (patch) | |
| tree | b237d917755a586c3dbe3fdb109901b9d0d6ef45 /tensorflow/compiler/xla/service/hlo_proto_util.h | |
| parent | eb690b51d1a7fb0e14884a5074c545295b6c2b23 (diff) | |
Add a convenient function that makes an HloProto from an HloModule.
PiperOrigin-RevId: 182438633
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_proto_util.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_proto_util.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_proto_util.h b/tensorflow/compiler/xla/service/hlo_proto_util.h index 603259a11f..320288fdb9 100644 --- a/tensorflow/compiler/xla/service/hlo_proto_util.h +++ b/tensorflow/compiler/xla/service/hlo_proto_util.h @@ -31,6 +31,10 @@ namespace xla { HloProto MakeHloProto(const HloModule& module, const BufferAssignment& assignment); +// Returns a serialized representation of the HLO state, but buffer assignment +// will not be included in the output. +HloProto MakeHloProto(const HloModule& module); + } // namespace xla #endif // TENSORFLOW_COMPILER_XLA_SERVICE_HLO_PROTO_UTIL_H_ |