diff --git a/tensorflow/core/kernels/tile_ops.cc b/tensorflow/core/kernels/tile_ops.cc index d714876bdaa964a35c9f011e34b6ec1d7b962ce7..b9b37612ad569fa8c23f4bb06d641a8c9215383d 100644 --- a/tensorflow/core/kernels/tile_ops.cc +++ b/tensorflow/core/kernels/tile_ops.cc @@ -325,6 +325,7 @@ class TileGradientOp : public OpKernel { TF_CALL_int16(HANDLE_TYPE_NAME); TF_CALL_int64(HANDLE_TYPE_NAME); TF_CALL_half(HANDLE_TYPE_NAME); + TF_CALL_bfloat16(HANDLE_TYPE_NAME); TF_CALL_complex64(HANDLE_TYPE_NAME); TF_CALL_complex128(HANDLE_TYPE_NAME);