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

Fixed compilation of elementwise_test.cc

Browse Source 
Reduced epsilon for FP16 in Exp test.

PiperOrigin-RevId: 302460743
Change-Id: I5e4e46e45181ad49e009b7b51e25bd9706aec936
This commit is contained in:
Raman Sarokin 2020-03-23 10:14:57 -07:00 committed by TensorFlower Gardener
parent 00ad1522c8
commit b71e31d545
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/delegates/gpu/cl/kernels/elementwise_test.cc
View File

@ -88,7 +88,7 @@ TEST_F(OpenCLOperationTest, Exp) {
for (auto storage : env_.GetSupportedStorages()) {
for (auto precision : env_.GetSupportedPrecisions()) {
const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f;
const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-2f;
OperationDef op_def;
op_def.precision = precision;
auto data_type = DeduceDataTypeFromPrecision(precision);
@ -499,7 +499,7 @@ TEST_F(OpenCLOperationTest, MaxiumumWithScalar) {
TensorFloat32 dst_tensor;
BroadcastSettings broadcast;
ElementwiseTwoInput operation = CreateElementwiseTwoInput(
creation_context_, op_def, OperationType::MAXIMUM, broadcast, attr);
creation_context_, op_def, OperationType::MAXIMUM, broadcast, &attr);
ASSERT_OK(ExecuteGPUOperation(src_tensor_0, creation_context_, &operation,
BHWC(1, 4, 1, 1), &dst_tensor));
EXPECT_THAT(dst_tensor.data,
@ -555,7 +555,7 @@ TEST_F(OpenCLOperationTest, MinimumWithScalar) {
TensorFloat32 dst_tensor;
BroadcastSettings broadcast;
ElementwiseTwoInput operation = CreateElementwiseTwoInput(
creation_context_, op_def, OperationType::MINIMUM, broadcast, attr);
creation_context_, op_def, OperationType::MINIMUM, broadcast, &attr);
ASSERT_OK(ExecuteGPUOperation(src_tensor_0, creation_context_, &operation,
BHWC(1, 4, 1, 1), &dst_tensor));
EXPECT_THAT(dst_tensor.data,