[TPU] Add option to only compile a replicated graph.
authorJacques Pienaar <jpienaar@google.com>
Sat, 5 May 2018 19:02:32 +0000 (12:02 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 7 May 2018 23:20:05 +0000 (16:20 -0700)
commita5e1809ddedbff8d9f66b82f9cf0989976694050
tree0a85466999be93f08cb49df5ce4480b612ecb5f5
parent62ed0aa37099e07720880a72a285304d34512cba
[TPU] Add option to only compile a replicated graph.

Useful when wanting to compile a computation but not run it. Returns a serialized CompilationResult string with the error message.

PiperOrigin-RevId: 195547847
tensorflow/contrib/tpu/BUILD
tensorflow/contrib/tpu/ops/replication_ops.cc
tensorflow/contrib/tpu/proto/BUILD
tensorflow/contrib/tpu/proto/compilation_result.proto [new file with mode: 0644]
tensorflow/contrib/tpu/python/tpu/tpu.py