(self)
| 664 | self.assertAllEqual([[15.0]], product) |
| 665 | |
| 666 | def testExecuteStringAttr(self): |
| 667 | checked_three = execute( |
| 668 | b'CheckNumerics', |
| 669 | num_outputs=1, |
| 670 | inputs=[constant_op.constant(3.)], |
| 671 | attrs=('message', 'just checking', 'T', |
| 672 | dtypes.float32.as_datatype_enum))[0] |
| 673 | self.assertEqual([[3]], checked_three.numpy()) |
| 674 | |
| 675 | def testExecuteStringAttrBadValue(self): |
| 676 | with self.assertRaises(errors.InvalidArgumentError): |