diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-09 16:40:03 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-09 16:42:52 -0700 |
commit | c07b719ab030c46f19c8e5cdd92730eaec38a8fb (patch) | |
tree | ac03b54d62c68ce528027e7e88d912618b8dac0e /tensorflow/compiler/xla/service/hlo_computation.h | |
parent | b348209171a2fac38def122d2ee43bd2fc3d9b1d (diff) |
[XLA] Make hlo deserialization stable for HloModule by sorting by ids when creating from proto.
Also, delete the HloModule parameter HloInstruction::CreateFromProto, it's not used anywhere.
Also, in ToProto, set sharding to proto if there is sharding.
PiperOrigin-RevId: 196049173
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_computation.h')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_computation.h | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_computation.h b/tensorflow/compiler/xla/service/hlo_computation.h index 9898355625..ba9d44a9ab 100644 --- a/tensorflow/compiler/xla/service/hlo_computation.h +++ b/tensorflow/compiler/xla/service/hlo_computation.h @@ -157,14 +157,12 @@ class HloComputation { // Creates a computation from the given proto. Arguments: // - // module: the module which will contain the computation. The newly created - // computation is *not* added to the module, however. // proto: the proto to convert from. // computation_map: a map from computation id to HloComputation*. This map // must contain all computations which the newly constructed computation // calls. static StatusOr<std::unique_ptr<HloComputation>> CreateFromProto( - HloModule* module, const HloComputationProto& proto, + const HloComputationProto& proto, const tensorflow::gtl::FlatMap<int64, HloComputation*>& computation_map); // Gets the instructions in this computation. |