diff options
| author |  Sanjoy Das <sanjoy@google.com> | 2018-02-16 15:29:35 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-16 15:33:12 -0800 |
| commit | ba019dc689d6393d8dba04ca57e8b01b374db14f (patch) | |
| tree | 6bd132cd6a1d3b6c8c833cb3e575db571ebd19a1 /tensorflow/compiler/xla/service/hlo_verifier.h | |
| parent | 1873ed4faab980ad239c06e8b92b8f4a85154fe3 (diff) | |
[XLA] Add some plumbing, documentation, verification and shape inference for Gather
Pretty much everything other than HLO verification and shape inference will fail
for Gather with Unimplemented.
Note that this CL is intentionally incomplete -- I figured it would be nicer to
get some of the boiler-platey stuff out of the way early. Let me know if you
want me to send in a larger but more complete CL instead.
PiperOrigin-RevId: 186055521
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_verifier.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_verifier.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_verifier.h b/tensorflow/compiler/xla/service/hlo_verifier.h index f9f898c236..1dd7ec3c51 100644 --- a/tensorflow/compiler/xla/service/hlo_verifier.h +++ b/tensorflow/compiler/xla/service/hlo_verifier.h @@ -80,6 +80,7 @@ class ShapeVerifier : public DfsHloVisitor { Status HandleBatchNormInference( HloInstruction* batch_norm_inference) override; Status HandleBatchNormGrad(HloInstruction* batch_norm_grad) override; + Status HandleGather(HloInstruction* gather) override; Status FinishVisit(HloInstruction*) override { return tensorflow::Status::OK(); |