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

Replace powf with pow, and atan2f with atan2.

Browse Source 
gcc doesn't support them.
Also add the missing header include <cmath>

PiperOrigin-RevId: 294625320
Change-Id: I3ccef02d38d8631bc47161ef1037cb47d5ec9cb6
This commit is contained in:
Adrian Kuegel 2020-02-12 02:44:17 -08:00 committed by TensorFlower Gardener
parent 8d154edcd5
commit f8070e16a3
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/xla/tests/exhaustive_binary_16_bit_test.cc
View File

@ -13,6 +13,8 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#include <cmath>
#include "tensorflow/compiler/xla/tests/exhaustive_op_test_utils.h"
#ifdef __FAST_MATH__
@ -122,11 +124,11 @@ BINARY_TEST_16BIT(Min, {
// TODO(bixia): Pow fails with bfloat16 on CPU.
BINARY_TEST_16BIT(DISABLED_ON_CPU(Pow),
{ Run(AddEmptyBroadcastDimension(Pow), std::powf); })
{ Run(AddEmptyBroadcastDimension(Pow), std::pow); })
// TODO(bixia): Atan2 fails with bfloat16 on CPU.
BINARY_TEST_16BIT(DISABLED_ON_CPU(Atan2),
{ Run(AddEmptyBroadcastDimension(Atan2), std::atan2f); })
{ Run(AddEmptyBroadcastDimension(Atan2), std::atan2); })
#if !defined(XLA_BACKEND_DOES_NOT_SUPPORT_FLOAT16)
INSTANTIATE_TEST_SUITE_P(F16, ExhaustiveF16BinaryTest,