Remove run_v1_only from experimental lite Python tests.
PiperOrigin-RevId: 261130125
This commit is contained in:
parent
ba775ddd39
commit
a5ca108de4
@ -22,7 +22,6 @@ import tensorflow as tf
|
|||||||
|
|
||||||
from tensorflow.lite.experimental.microfrontend.python.ops import audio_microfrontend_op as frontend_op
|
from tensorflow.lite.experimental.microfrontend.python.ops import audio_microfrontend_op as frontend_op
|
||||||
from tensorflow.python.framework import ops
|
from tensorflow.python.framework import ops
|
||||||
from tensorflow.python.framework import test_util
|
|
||||||
|
|
||||||
SAMPLE_RATE = 1000
|
SAMPLE_RATE = 1000
|
||||||
WINDOW_SIZE = 25
|
WINDOW_SIZE = 25
|
||||||
@ -57,7 +56,6 @@ class AudioFeatureGenerationTest(tf.test.TestCase):
|
|||||||
self.assertAllEqual(filterbanks.eval(),
|
self.assertAllEqual(filterbanks.eval(),
|
||||||
[[479, 425], [436, 378], [410, 350], [391, 325]])
|
[[479, 425], [436, 378], [410, 350], [391, 325]])
|
||||||
|
|
||||||
@test_util.run_v1_only("b/120545219")
|
|
||||||
def testSimpleFloatScaled(self):
|
def testSimpleFloatScaled(self):
|
||||||
with self.test_session():
|
with self.test_session():
|
||||||
audio = tf.constant(
|
audio = tf.constant(
|
||||||
@ -79,7 +77,6 @@ class AudioFeatureGenerationTest(tf.test.TestCase):
|
|||||||
[[7.484375, 6.640625], [6.8125, 5.90625],
|
[[7.484375, 6.640625], [6.8125, 5.90625],
|
||||||
[6.40625, 5.46875], [6.109375, 5.078125]])
|
[6.40625, 5.46875], [6.109375, 5.078125]])
|
||||||
|
|
||||||
@test_util.run_v1_only("b/120545219")
|
|
||||||
def testStacking(self):
|
def testStacking(self):
|
||||||
with self.test_session():
|
with self.test_session():
|
||||||
audio = tf.constant(
|
audio = tf.constant(
|
||||||
@ -122,7 +119,6 @@ class AudioFeatureGenerationTest(tf.test.TestCase):
|
|||||||
[[479, 425, 479, 425, 436, 378], [479, 425, 436, 378, 410, 350],
|
[[479, 425, 479, 425, 436, 378], [479, 425, 436, 378, 410, 350],
|
||||||
[436, 378, 410, 350, 391, 325], [410, 350, 391, 325, 391, 325]])
|
[436, 378, 410, 350, 391, 325], [410, 350, 391, 325, 391, 325]])
|
||||||
|
|
||||||
@test_util.run_v1_only("b/120545219")
|
|
||||||
def testStackingDropFrame(self):
|
def testStackingDropFrame(self):
|
||||||
with self.test_session():
|
with self.test_session():
|
||||||
audio = tf.constant(
|
audio = tf.constant(
|
||||||
|
Loading…
Reference in New Issue
Block a user