Mirrored strategy and default strategy can result in different layouts in XLA that can result in different accumulation orders for reduction operations. As a result, relax the precision of the keras correctness tests.
PiperOrigin-RevId: 327474361 Change-Id: I8fcf1a2889d6c99f5c757fc8f6a57d1f20cb973f
This commit is contained in:
parent
bbcacea880
commit
800b502f00
@ -308,7 +308,7 @@ def compare_results(results_with_ds,
|
||||
default_tolerance = 1e-3
|
||||
relaxed_tolerance = 1e-3
|
||||
else:
|
||||
default_tolerance = 4e-5
|
||||
default_tolerance = 1e-5
|
||||
relaxed_tolerance = 1e-4
|
||||
|
||||
def _get_compare_result_tolerance(key):
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user