From fa12754826ca8502e03ef7b7519e734558300a3f Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 4 Mar 2021 14:17:06 -0800
Subject: [PATCH] Add new data transfer support.

PiperOrigin-RevId: 361002139
Change-Id: I779bfffeb8659e2d31381cdffdacc2cda591081b
---
 tensorflow/compiler/xla/service/hlo_runner_interface.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tensorflow/compiler/xla/service/hlo_runner_interface.h b/tensorflow/compiler/xla/service/hlo_runner_interface.h
index aeddd31192c..071d4b5e066 100644
--- a/tensorflow/compiler/xla/service/hlo_runner_interface.h
+++ b/tensorflow/compiler/xla/service/hlo_runner_interface.h
@@ -169,6 +169,8 @@ class HloRunnerInterface {
       std::function<int64(int64)> argument_count_provider,
       std::function<const Literal*(int64, int64)> argument_provider,
       const ReplicatedExecuteOptions& options) = 0;
+
+  typedef std::function<Shape(const Shape&)> TpuShapeRepresentationFn;
 };
 
 }  // namespace xla