Add methods for extracting the shapes of the entry computation from an HloProto.
authorMark Heffernan <meheff@google.com>
Mon, 5 Mar 2018 21:44:42 +0000 (13:44 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 5 Mar 2018 21:48:53 +0000 (13:48 -0800)
commit36b3c94a99704c8e1973ae5c043aec4870ae84ff
tree1b40289dbc0d6de7ec8ff3e5f15823754c867bd3
parent5368a1a3af94c6b49dd51d0d85cb3702f484daa7
Add methods for extracting the shapes of the entry computation from an HloProto.

PiperOrigin-RevId: 187915821
tensorflow/compiler/xla/service/BUILD
tensorflow/compiler/xla/service/hlo_proto_util.cc
tensorflow/compiler/xla/service/hlo_proto_util.h
tensorflow/compiler/xla/service/hlo_proto_util_test.cc [new file with mode: 0644]