Skip to content

Commit df472f4

Browse files
authored
Fix ut 1.15up3 (#846)
* fix mixed precision not support tf1.15UP3
1 parent 8f39331 commit df472f4

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
intel-tensorflow==1.15.2
1+
https://storage.googleapis.com/intel-optimized-tensorflow/intel_tensorflow-1.15.0up3-cp37-cp37m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
22
tf_slim

test/test_bf16_convert.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -290,6 +290,7 @@ def test_do_transform(self):
290290
self.assertEqual(new_relu2.attr["T"].type, dtypes.bfloat16)
291291
self.assertTrue("relu2_BF16toFP32" in new_conv3.input)
292292

293+
293294
@unittest.skipIf(tf.version.VERSION > '2.5.0', " Skip test_bf16_fallback case for tf 2.6.0 and above.")
294295
def test_bf16_fallback(self):
295296
os.environ['FORCE_BF16'] = '1'

0 commit comments

Comments
 (0)