Unable to build time_two_op op using Makefile in Docker.

[sebastian-sz]

To reproduce:

1. git clone, cd and Run docker: docker run -it --rm --gpus all -w /workspace/ -v ${PWD}:/workspace/ tensorflow/tensorflow:custom-op-gpu-ubuntu16
2. Attempt to build the op: make time_two_op

Will result in

2022-04-08 10:56:26.179690: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0
2022-04-08 10:56:27.401461: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0
nvcc -std=c++11 -c -o tensorflow_time_two/python/ops/_time_two_ops.cu.o tensorflow_time_two/cc/kernels/time_two_kernels.cu.cc  -I/usr/local/lib/python3.6/dist-packages/tensorflow/include -D_GLIBCXX_USE_CXX11_ABI=0 -D GOOGLE_CUDA=1 -x cu -Xcompiler -fPIC -DNDEBUG --expt-relaxed-constexpr
In file included from tensorflow_time_two/cc/kernels/time_two_kernels.cu.cc:21:0:
/usr/local/lib/python3.6/dist-packages/tensorflow/include/tensorflow/core/util/gpu_kernel_helper.h:24:53: fatal error: third_party/gpus/cuda/include/cuda_fp16.h: No such file or directory
compilation terminated.