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