Implement error code in TFRT and propagate proper error code to c_api_tfrt.
PiperOrigin-RevId: 334529554 Change-Id: I96c058f8971b150d2cbb9ee7e9ba91c718a4b505
This commit is contained in:
parent
a9ef1af36b
commit
e79e90cb04
@ -581,13 +581,11 @@ class FillTest(test.TestCase):
|
|||||||
tf_ans = array_ops.fill([2, 3], np_ans[0][0], name="fill").numpy()
|
tf_ans = array_ops.fill([2, 3], np_ans[0][0], name="fill").numpy()
|
||||||
self.assertAllEqual(np_ans, tf_ans)
|
self.assertAllEqual(np_ans, tf_ans)
|
||||||
|
|
||||||
@test_util.disable_tfrt("support error code in TFRT.")
|
|
||||||
def testFillNegative(self):
|
def testFillNegative(self):
|
||||||
for shape in (-1,), (2, -1), (-1, 2), (-2), (-3):
|
for shape in (-1,), (2, -1), (-1, 2), (-2), (-3):
|
||||||
with self.assertRaises(errors_impl.InvalidArgumentError):
|
with self.assertRaises(errors_impl.InvalidArgumentError):
|
||||||
array_ops.fill(shape, 7)
|
array_ops.fill(shape, 7)
|
||||||
|
|
||||||
@test_util.disable_tfrt("support error code in TFRT.")
|
|
||||||
def testShapeFunctionEdgeCases(self):
|
def testShapeFunctionEdgeCases(self):
|
||||||
# Non-vector dimensions.
|
# Non-vector dimensions.
|
||||||
with self.assertRaises(errors_impl.InvalidArgumentError):
|
with self.assertRaises(errors_impl.InvalidArgumentError):
|
||||||
|
Loading…
x
Reference in New Issue
Block a user