Lines Matching refs:tensor_util
15 """Functional tests for tensor_util."""
28 from tensorflow.python.framework import tensor_util
41 t = tensor_util.make_tensor_proto(value)
47 a = tensor_util.MakeNdarray(t)
52 t = tensor_util.make_tensor_proto([10.0, 20.0, 30.0])
65 a = tensor_util.MakeNdarray(t)
70 t = tensor_util.make_tensor_proto([10.0, 20.0, 30.0], dtype=dtypes.float32)
83 a = tensor_util.MakeNdarray(t)
88 t = tensor_util.make_tensor_proto([10, 20, 30], dtype=dtypes.float32)
101 a = tensor_util.MakeNdarray(t)
107 t = tensor_util.make_tensor_proto(arr, dtype=dtypes.float32)
120 a = tensor_util.MakeNdarray(t)
125 t = tensor_util.make_tensor_proto([10.0, 20.0, 30.0], shape=[1, 3])
138 a = tensor_util.MakeNdarray(t)
143 t = tensor_util.make_tensor_proto([10.0, 20.0, 30.0], shape=[3, 1])
156 a = tensor_util.MakeNdarray(t)
161 t = tensor_util.make_tensor_proto(10.0, shape=[1, 3])
170 t = tensor_util.make_tensor_proto(
184 a = tensor_util.MakeNdarray(t)
188 tensor_util.MakeNdarray(t))
193 t = tensor_util.make_tensor_proto([10.0], shape=[3, 4], dtype=dtype)
194 a = tensor_util.MakeNdarray(t)
204 t = tensor_util.make_tensor_proto([10.0, 20.0, 30.0], dtype=dtypes.float32)
205 a = tensor_util.MakeNdarray(t)
223 t = tensor_util.make_tensor_proto(np.array([10.0, 20.0], dtype=np.float16))
235 a = tensor_util.MakeNdarray(t)
241 t = tensor_util.make_tensor_proto(np.array([10.0, 20.0], dtype=test_type))
255 a = tensor_util.MakeNdarray(t)
260 t = tensor_util.make_tensor_proto(10)
266 a = tensor_util.MakeNdarray(t)
272 t = tensor_util.make_tensor_proto(value)
278 a = tensor_util.MakeNdarray(t)
290 t = tensor_util.make_tensor_proto(value)
296 a = tensor_util.MakeNdarray(t)
301 t = tensor_util.make_tensor_proto([10, 20, 30, 40], shape=[2, 2])
314 a = tensor_util.MakeNdarray(t)
325 t = tensor_util.make_tensor_proto([10, 20, 30], dtype=dtype)
328 a = tensor_util.MakeNdarray(t)
332 t = tensor_util.make_tensor_proto(np.array([10, 20, 30], dtype=nptype))
335 a = tensor_util.MakeNdarray(t)
346 tensor_util.MakeNdarray(
347 tensor_util.make_tensor_proto([10, 11, 12],
355 t = tensor_util.make_tensor_proto(
358 a = tensor_util.MakeNdarray(t)
363 t = tensor_util.make_tensor_proto(10, dtype=dtypes.int64)
369 a = tensor_util.MakeNdarray(t)
374 t = tensor_util.make_tensor_proto(
388 a = tensor_util.MakeNdarray(t)
393 t = tensor_util.make_tensor_proto(np.array([10, 20, 30]))
406 a = tensor_util.MakeNdarray(t)
414 t = tensor_util.make_tensor_proto(data, dtype=dtypes.qint32)
427 a = tensor_util.MakeNdarray(t)
431 t = tensor_util.make_tensor_proto(data, dtype=dtypes.quint8)
437 a = tensor_util.MakeNdarray(t)
441 t = tensor_util.make_tensor_proto(data, dtype=dtypes.qint8)
447 tensor_util.MakeNdarray(t)
451 t = tensor_util.make_tensor_proto(data, dtype=dtypes.quint16)
464 a = tensor_util.MakeNdarray(t)
468 t = tensor_util.make_tensor_proto(data, dtype=dtypes.qint16)
481 a = tensor_util.MakeNdarray(t)
486 t = tensor_util.make_tensor_proto("foo")
492 a = tensor_util.MakeNdarray(t)
497 t = tensor_util.make_tensor_proto(["f", "g"], shape=[3, 4])
498 a = tensor_util.MakeNdarray(t)
505 t = tensor_util.make_tensor_proto([b"foo", b"bar", b"baz"], shape=[1, 3])
513 a = tensor_util.MakeNdarray(t)
518 t = tensor_util.make_tensor_proto(
528 a = tensor_util.MakeNdarray(t)
539 t = tensor_util.make_tensor_proto(Wrapper(), shape=[1, 3])
547 a = tensor_util.MakeNdarray(t)
559 t = tensor_util.make_tensor_proto(Wrapper(), shape=[1, 3])
567 a = tensor_util.MakeNdarray(t)
572 t = tensor_util.make_tensor_proto((b"a", b"ab", b"abc", b"abcd"))
581 a = tensor_util.MakeNdarray(t)
586 t = tensor_util.make_tensor_proto(((b"a", b"ab"), (b"abc", b"abcd")))
595 a = tensor_util.MakeNdarray(t)
600 t = tensor_util.make_tensor_proto((1 + 2j), dtype=dtypes.complex64)
607 a = tensor_util.MakeNdarray(t)
612 t = tensor_util.make_tensor_proto((1 + 2j), dtype=dtypes.complex128)
619 a = tensor_util.MakeNdarray(t)
626 t = tensor_util.make_tensor_proto((1 + 1j), shape=[3, 4], dtype=dtype)
627 a = tensor_util.MakeNdarray(t)
637 t = tensor_util.make_tensor_proto(
649 a = tensor_util.MakeNdarray(t)
654 t = tensor_util.make_tensor_proto(
666 a = tensor_util.MakeNdarray(t)
671 t = tensor_util.make_tensor_proto(
687 a = tensor_util.MakeNdarray(t)
693 t = tensor_util.make_tensor_proto(
709 a = tensor_util.MakeNdarray(t)
716 tensor_util.make_tensor_proto(np.array([1]), 0)
718 tensor_util.make_tensor_proto(3, dtype=dtypes.qint8)
720 tensor_util.make_tensor_proto([3], dtype=dtypes.qint8)
725 tensor_util.make_tensor_proto([tensor_shape.Dimension(1)])
731 tensor_util.make_tensor_proto(array, shape=correct_shape, verify_shape=True)
733 tensor_util.make_tensor_proto(
738 tensor_util.make_tensor_proto(np.array([1, 2]), shape=[1])
741 t = tensor_util.make_tensor_proto(np.array(7))
749 t = tensor_util.make_tensor_proto([10, 20, 30, 40], shape=[2, 2])
750 self.assertTrue(tensor_util.ShapeEquals(t, [2, 2]))
751 self.assertTrue(tensor_util.ShapeEquals(t, (2, 2)))
753 tensor_util.ShapeEquals(t, tensor_shape.as_shape([2, 2]).as_proto()))
754 self.assertFalse(tensor_util.ShapeEquals(t, [5, 3]))
755 self.assertFalse(tensor_util.ShapeEquals(t, [1, 4]))
756 self.assertFalse(tensor_util.ShapeEquals(t, [4]))
783 self.assertFalse(tensor_util.is_tensor(np_val))
784 self.assertTrue(tensor_util.is_tensor(tf_val))
792 self.assertAllClose(np_val, tensor_util.constant_value(tf_val))
796 self.assertAllClose(np_val, tensor_util.constant_value(tf_val))
806 self.assertIs(None, tensor_util.constant_value(tf_val))
811 c_val = tensor_util.constant_value(tf_val)
818 c_val = tensor_util.constant_value(tf_val)
824 c_val = tensor_util.constant_value(tf_val)
830 c_val = tensor_util.constant_value(tf_val)
837 c_val = tensor_util.constant_value(tf_val)
848 c_val = tensor_util.constant_value(tf_val)
858 c_val = tensor_util.constant_value(tf_val)
863 c_val = tensor_util.constant_value(tf_val)
871 c_val = tensor_util.constant_value(tf_val)
877 c_val = tensor_util.constant_value(tf_val)
883 c_val = tensor_util.constant_value(tf_val)
891 c_val = tensor_util.constant_value(tf_val)
896 c_val = tensor_util.constant_value(tf_val)
903 c_val = tensor_util.constant_value(tf_val)
908 c_val = tensor_util.constant_value(tf_val)
915 c_val = tensor_util.constant_value(tf_val, partial=True)
924 c_val = tensor_util.constant_value(tf_val, partial=True)
930 self.assertEqual(tensor_util.constant_value(tf_val), True)
933 self.assertEqual(tensor_util.constant_value(tf_val), False)
938 c_val = tensor_util.constant_value(tf_val)
945 self.assertEqual(tensor_util.constant_value(tf_val), False)
949 self.assertEqual(tensor_util.constant_value(tf_val), True)
954 c_val = tensor_util.constant_value(tf_val)
959 self.assertIs(x, tensor_util.constant_value(x))
963 self.assertIs(np_val, tensor_util.constant_value(np_val))
967 self.assertIsNone(tensor_util.constant_value(var))
971 self.assertIsNone(tensor_util.constant_value(var))
982 tensor_util.constant_value_as_shape(tf_val))
987 tensor_util.constant_value_as_shape(tf_val))
992 c_val = tensor_util.constant_value_as_shape(tf_val)
998 c_val = tensor_util.constant_value_as_shape(tf_val)
1005 c_val = tensor_util.constant_value_as_shape(tf_val)
1013 c_val = tensor_util.constant_value_as_shape(tf_val)
1019 c_val = tensor_util.constant_value_as_shape(tf_val)
1025 c_val = tensor_util.constant_value_as_shape(tf_val)
1030 c_val = tensor_util.constant_value_as_shape(tf_val)
1036 c_val = tensor_util.constant_value_as_shape(tf_val)
1046 c_val = tensor_util.constant_value_as_shape(tf_val)
1051 c_val = tensor_util.constant_value_as_shape(tf_val)
1056 c_val = tensor_util.constant_value_as_shape(tf_val)
1061 c_val = tensor_util.constant_value_as_shape(tf_val)
1066 c_val = tensor_util.constant_value_as_shape(tf_val)
1071 c_val = tensor_util.constant_value_as_shape(tf_val)
1076 c_val = tensor_util.constant_value_as_shape(tf_val)
1081 c_val = tensor_util.constant_value_as_shape(tf_val)
1086 c_val = tensor_util.constant_value_as_shape(tf_val)
1092 c_val = tensor_util.constant_value_as_shape(tf_val)
1098 c_val = tensor_util.constant_value_as_shape(tf_val)
