From b813277fd5121698dd243091d2778dd12daa13e5 Mon Sep 17 00:00:00 2001 From: Allen Wang Date: Thu, 4 Mar 2021 21:39:43 -0800 Subject: [PATCH] Include `call_once.h` in `tpu_functional_ops.h` PiperOrigin-RevId: 361070186 Change-Id: I0d8fe8950744119cab53a94c8d1116385380f37e --- tensorflow/core/tpu/kernels/tpu_functional_ops.h | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/core/tpu/kernels/tpu_functional_ops.h b/tensorflow/core/tpu/kernels/tpu_functional_ops.h index 7d5339c175a..c60b9fe7d7c 100644 --- a/tensorflow/core/tpu/kernels/tpu_functional_ops.h +++ b/tensorflow/core/tpu/kernels/tpu_functional_ops.h @@ -16,6 +16,7 @@ limitations under the License. #ifndef TENSORFLOW_CORE_TPU_KERNELS_TPU_FUNCTIONAL_OPS_H_ #define TENSORFLOW_CORE_TPU_KERNELS_TPU_FUNCTIONAL_OPS_H_ +#include "absl/base/call_once.h" #include "tensorflow/compiler/jit/shape_inference.h" #include "tensorflow/core/common_runtime/device_mgr.h" #include "tensorflow/core/common_runtime/optimization_registry.h"