Skip to content

Commit e8219d1

Browse files
Automated Code Change
PiperOrigin-RevId: 765059040
1 parent 3d22e3f commit e8219d1

4 files changed

Lines changed: 8 additions & 1 deletion

File tree

tensorflow/core/kernels/image/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ tf_kernel_library(
213213
tf_kernel_library(
214214
name = "draw_bounding_box_op",
215215
prefix = "draw_bounding_box_op",
216-
deps = IMAGE_DEPS,
216+
deps = IMAGE_DEPS + ["@com_google_absl//absl/log"],
217217
)
218218

219219
tf_kernel_library(

tensorflow/core/kernels/image/attention_ops.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ limitations under the License.
1515

1616
// See docs in ../ops/image_ops.cc.
1717

18+
#include <cstdint>
1819
#define EIGEN_USE_THREADS
1920

2021
#include <vector>

tensorflow/core/kernels/image/draw_bounding_box_op.cc

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,11 @@ See the License for the specific language governing permissions and
1313
limitations under the License.
1414
==============================================================================*/
1515
// See ../ops/image_ops.cc for details.
16+
#include <algorithm>
17+
#include <cstdint>
18+
#include <vector>
19+
20+
#include "absl/log/log.h"
1621
#define EIGEN_USE_THREADS
1722

1823
#include "tensorflow/core/framework/op_kernel.h"

tensorflow/core/kernels/image/resize_nearest_neighbor_op_gpu.cu.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ See the License for the specific language governing permissions and
1313
limitations under the License.
1414
==============================================================================*/
1515

16+
#include <cmath>
1617
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
1718

1819
#define EIGEN_USE_GPU

0 commit comments

Comments
 (0)