This fixes build warnings (*see example below) introduced in my previous random ops merge request.
The referencing of undefined math functions is suspected to cause test failures on AMD GPUs.
17:20:11 INFO: From Compiling tensorflow/core/kernels/random_op.cc:
17:20:11 warning: [Computecpp:CC0034]: Function sqrt is undefined but referenced on the device and the associated kernels may fail to build or execute at run time [-Wsycl-undef-func]
17:20:11 warning: [Computecpp:CC0034]: Function log is undefined but referenced on the device and the associated kernels may fail to build or execute at run time [-Wsycl-undef-func]
...
This fixes build warnings (*see example below) introduced in my previous random ops merge request. The referencing of undefined math functions is suspected to cause test failures on AMD GPUs.
For a full list see here