Class XlaComputation

tensorflow/compiler/xla/client/xla_computation.h:29–64

The computation graph that the user builds up with the XlaBuilder.

// The computation graph that the user builds up with the XlaBuilder.
class XlaComputation {
 public:
  XlaComputation() : unique_id_(-1) {}
  XlaComputation(const HloModuleProto& proto)
      : unique_id_(proto.id()), proto_(proto) {}

  ~XlaComputation() {}

  XlaComputation(const XlaComputation&) = delete;
  XlaComputation& operator=(const XlaComputation&) = delete;

  XlaComputation(XlaComputation&& from) = default;

  XlaComputation& operator=(XlaComputation&& from) = default;

  // Returns the "program shape" (parameter and return shapes) for this
  // computation.
  StatusOr<ProgramShape> GetProgramShape() const;

  const HloModuleProto& proto() const { return proto_; }

  // Requests that we snapshot the computation into a serializable protocol
  // buffer form.
  StatusOr<std::unique_ptr<HloSnapshot>> Snapshot() const;

  // Returns true if this object is a null Computation.
  bool IsNull() const { return unique_id_ == -1; }

 private:
  XlaComputation(const int64 unique_id) : unique_id_(unique_id) {}
  HloModuleProto* mutable_proto() { return &proto_; }
  friend class XlaBuilder;

  int64 unique_id_;
  HloModuleProto proto_;
};

}  // namespace xla