From e0575253d11c6a57bc25ddfec09d4d2e1f2a47c1 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 19 Feb 2020 10:11:55 -0800
Subject: [PATCH] Provide an accessor for dynamic dimension inference.

PiperOrigin-RevId: 295996157
Change-Id: I50ea04cd692d1163b2e05d9f8e12dbeffc11fa3d
---
 tensorflow/compiler/xla/service/hlo_evaluator.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tensorflow/compiler/xla/service/hlo_evaluator.h b/tensorflow/compiler/xla/service/hlo_evaluator.h
index fc9d42c1b17..803004225d2 100644
--- a/tensorflow/compiler/xla/service/hlo_evaluator.h
+++ b/tensorflow/compiler/xla/service/hlo_evaluator.h
@@ -133,6 +133,10 @@ class HloEvaluator : public DfsHloVisitorWithDefault {
     dynamic_dimension_inference_ = dynamic_dimension_inference;
   }
 
+  DynamicDimensionInference* dynamic_dimension_inference() {
+    return dynamic_dimension_inference_;
+  }
+
   // Enable the fast path for certain operations like dot or convolution.
   void set_use_fast_path(bool value) { use_fast_path_ = value; }