diff options
author | 2018-04-09 22:04:04 -0700 | |
---|---|---|
committer | 2018-04-09 22:06:41 -0700 | |
commit | 022d1fe0ecb46a2a7b77f8b99de8c273ef804c82 (patch) | |
tree | 87b2813a8b54e41a9424a97a18eb386781de3a4a /tensorflow/compiler/xla/xla.proto | |
parent | 3f3a6e6685449130389e0b8d76f6ba0fc457bcfe (diff) |
[XLA] Redesign: implement XlaBuilder::IsConstant, XlaBuidler::BuildConstantSubGraph, and Client::ComputeConstant(XlaComputation...).
- Since the builder no longer holds a client, we moved the ComputeConstant to the client side so that it can communicate with the service side. Now we add XlaBuilder::BuildConstantSubGraph, which is only responsible for building a subgraph that is compile-time constant.
- Before this change, every XlaBuilder has a unique id. Now since it also builds constant subgraph, we give every XlaComputation being built a global unique id, and uniquify instruction names when actually building the XlaComputation.
PiperOrigin-RevId: 192236997
Diffstat (limited to 'tensorflow/compiler/xla/xla.proto')
-rw-r--r-- | tensorflow/compiler/xla/xla.proto | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/xla.proto b/tensorflow/compiler/xla/xla.proto index f9943f71d3..b4cbdf3773 100644 --- a/tensorflow/compiler/xla/xla.proto +++ b/tensorflow/compiler/xla/xla.proto @@ -417,6 +417,11 @@ message ComputeConstantRequest { repeated LiteralProto parameters = 4; } +message ComputeConstantGraphRequest { + HloModuleProto computation = 1; + Layout output_layout = 2; +} + message ComputeConstantResponse { // A LiteralProto is returned directly for this request, instead of a // ComputationDataHandle. |