diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/computation_placer.h')
-rw-r--r-- | tensorflow/compiler/xla/service/computation_placer.h | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/computation_placer.h b/tensorflow/compiler/xla/service/computation_placer.h index 4d26d6bb85..7d9abcd100 100644 --- a/tensorflow/compiler/xla/service/computation_placer.h +++ b/tensorflow/compiler/xla/service/computation_placer.h @@ -49,7 +49,11 @@ class DeviceAssignment : public Array2D<int> { // Protocol buffer serialization and deserialization. Status Serialize(DeviceAssignmentProto* proto) const; - static StatusOr<DeviceAssignment> Deserialize( + + // Return a std::unique_ptr<DeviceAssignment> instead of a DeviceAssignment + // directly because one of the supported TF platforms (mac) does not compile + // due to a StatusOr of an incomplete type (DeviceAssignment). + static StatusOr<std::unique_ptr<DeviceAssignment>> Deserialize( const DeviceAssignmentProto& proto); }; |