(self)
| 1007 | 6.0) |
| 1008 | |
| 1009 | def testMakeAttrShape(self): |
| 1010 | for s in ([], None, [1, 2, 3], [None, None], [1, None, 3]): |
| 1011 | expected = tensor_shape.TensorShape(s).as_proto() |
| 1012 | actual = backprop.make_attr(pywrap_tensorflow.TF_ATTR_SHAPE, s) |
| 1013 | self.assertEqual( |
| 1014 | expected, |
| 1015 | actual, |
| 1016 | msg=('For shape %r, expected %r != %r actual' % (s, expected, |
| 1017 | actual))) |
| 1018 | |
| 1019 | def testMakeAttrShapeList(self): |
| 1020 | shape_list = [[], None, [1, 2, 3], [None, None], [1, None, 3]] |