experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Include call_once.h in tpu_functional_ops.h

Browse Source 
PiperOrigin-RevId: 361070186
Change-Id: I0d8fe8950744119cab53a94c8d1116385380f37e
This commit is contained in:
Allen Wang 2021-03-04 21:39:43 -08:00 committed by TensorFlower Gardener
parent 2a5cbed7e9
commit b813277fd5
1 changed files with 1 additions and 0 deletions

1
tensorflow/core/tpu/kernels/tpu_functional_ops.h
View File

@ -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"