Home
last modified time | relevance | path

Searched refs:assertRaggedEqual (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_operators_test.py33 self.assertRaggedEqual((x > y), [[False, False], [True]])
34 self.assertRaggedEqual((x >= y), [[False, True], [True]])
35 self.assertRaggedEqual((x < y), [[True, False], [False]])
36 self.assertRaggedEqual((x <= y), [[True, True], [False]])
41 self.assertRaggedEqual(abs(x), [[1.0, 2.0], [8.0]])
43 self.assertRaggedEqual((-x), [[-1.0, 2.0], [-8.0]])
45 self.assertRaggedEqual((x + y), [[5.0, 2.0], [10.0]])
46 self.assertRaggedEqual((3.0 + y), [[7.0, 7.0], [5.0]])
47 self.assertRaggedEqual((x + 3.0), [[4.0, 1.0], [11.0]])
49 self.assertRaggedEqual((x - y), [[-3.0, -6.0], [6.0]])
[all …]
Dragged_range_op_test.py34 self.assertRaggedEqual(rt1, [[0, 1, 2], [0, 1, 2, 3, 4], [0, 1]])
37 self.assertRaggedEqual(rt2, [[0, 1, 2], [], [8, 9, 10, 11]])
40 self.assertRaggedEqual(rt3, [[0, 2], [], [8, 10]])
44 self.assertRaggedEqual(
50 self.assertRaggedEqual(
57 self.assertRaggedEqual(
73 self.assertRaggedEqual(
79 self.assertRaggedEqual(
87 self.assertRaggedEqual(
94 self.assertRaggedEqual(
[all …]
Dragged_tensor_bounding_shape_op_test.py35 self.assertRaggedEqual(rt.bounding_shape(), [5, 4])
42 self.assertRaggedEqual(rt1.bounding_shape(), [5, 3])
43 self.assertRaggedEqual(rt2.bounding_shape(), [1, 7])
44 self.assertRaggedEqual(rt3.bounding_shape(), [3, 7])
51 self.assertRaggedEqual(rt1.bounding_shape(), [5, 3, 2])
52 self.assertRaggedEqual(rt2.bounding_shape(), [1, 7, 2])
53 self.assertRaggedEqual(rt3.bounding_shape(), [3, 7, 2])
58 self.assertRaggedEqual(rt.bounding_shape(0), 5)
59 self.assertRaggedEqual(rt.bounding_shape(1), 3)
60 self.assertRaggedEqual(rt.bounding_shape([1, 0]), [3, 5])
Dragged_gather_op_test.py42 self.assertRaggedEqual(
45 self.assertRaggedEqual(
48 self.assertRaggedEqual(
55 self.assertRaggedEqual(
63 self.assertRaggedEqual(
70 self.assertRaggedEqual(
78 self.assertRaggedEqual(
89 self.assertRaggedEqual(
96 self.assertRaggedEqual(
110 self.assertRaggedEqual(
Dragged_from_tensor_op_test.py39 self.assertRaggedEqual(
42 self.assertRaggedEqual(
45 self.assertRaggedEqual(
51 self.assertRaggedEqual(
323 self.assertRaggedEqual(rt, expected)
337 self.assertRaggedEqual(rt, self.evaluate(dt).tolist())
447 self.assertRaggedEqual(rt, expected)
Dragged_map_fn_op_test.py171 self.assertRaggedEqual(expected_rt, output)
189 self.assertRaggedEqual(output, [66, 44, 198])
219 self.assertRaggedEqual(output['batman'], [[2, 3, 4], [5], [6, 7, 8]])
220 self.assertRaggedEqual(output['robin'], [[11, 21, 31], [41], [51, 61, 71]])
237 self.assertRaggedEqual(
258 self.assertRaggedEqual(
293 self.assertRaggedEqual(id_t2, [[0, 5], [0, 4]])
Dragged_map_flat_values_op_test.py44 self.assertRaggedEqual(result, expected)
52 self.assertRaggedEqual(v1, [[1, 1, 1], [], [1, 1], [1]])
53 self.assertRaggedEqual(v2, [[1, 4, 9], [], [16, 25], [36]])
54 self.assertRaggedEqual(v3, [[6, 7, 8], [], [9, 10], [11]])
125 self.assertRaggedEqual(
Dragged_segment_op_test.py113 self.assertRaggedEqual(segmented, expected)
164 self.assertRaggedEqual(segmented1, expected1)
171 self.assertRaggedEqual(segmented2, expected2)
185 self.assertRaggedEqual(segmented, expected)
Dragged_from_sparse_op_test.py42 self.assertRaggedEqual(rt, [[1, 2, 3], [4], [], [5]])
51 self.assertRaggedEqual(rt, [[], [], [], []])
Dragged_tile_op_test.py212 self.assertRaggedEqual(tiled, expected)
221 self.assertRaggedEqual(tiled, expected)
Dragged_to_sparse_op_test.py195 self.assertRaggedEqual(g1, [[1.0, 1.0], [1.0, 1.0], [1.0, 1.0]])
196 self.assertRaggedEqual(g2, [[2.0, 2.0], [2.0, 2.0], [2.0, 2.0]])
Dragged_eager_test.py39 self.assertRaggedEqual(rt, pylist)
Dragged_concat_op_test.py243 self.assertRaggedEqual(concatenated, expected)
321 self.assertRaggedEqual(concatenated, [[1, 2], [3, 4]])
Dragged_stack_op_test.py338 self.assertRaggedEqual(stacked, expected)
375 self.assertRaggedEqual(stacked, [[[1, 2], [3, 4]]])
Dragged_reduce_op_test.py311 self.assertRaggedEqual(reduced, expected)
331 self.assertRaggedEqual(reduced, expected)
Dragged_batch_gather_op_test.py152 self.assertRaggedEqual(result, expected)
362 self.assertRaggedEqual(result, expected)
Dragged_expand_dims_op_test.py123 self.assertRaggedEqual(expanded, expected)
Dragged_row_lengths_op_test.py123 self.assertRaggedEqual(lengths, expected)
Dragged_tensor_shape_test.py425 self.assertRaggedEqual(result, expected)
487 self.assertRaggedEqual(result, expected)
Dragged_test_util.py48 def assertRaggedEqual(self, a, b): member in RaggedTensorTestCase
Dragged_dispatch_test.py471 self.assertRaggedEqual(result, expected)
702 self.assertRaggedEqual(result, expected)
Dragged_where_op_test.py184 self.assertRaggedEqual(result, expected)
Dragged_gather_nd_op_test.py205 self.assertRaggedEqual(result, expected)
/external/tensorflow/tensorflow/python/kernel_tests/
Dunicode_encode_op_test.py36 def assertRaggedEqual(self, rt, expected): member in UnicodeEncodeOpTest
91 self.assertRaggedEqual(unicode_encode_op, expected_value)
98 self.assertRaggedEqual(unicode_encode_op, expected_value)
104 self.assertRaggedEqual(unicode_encode_op, expected_value)
121 self.assertRaggedEqual(unicode_encode_op, expected_value)
126 self.assertRaggedEqual(unicode_encode_op, expected_value)
132 self.assertRaggedEqual(unicode_encode_op, expected_value)
137 self.assertRaggedEqual(unicode_encode_op, expected_value)
161 self.assertRaggedEqual(unicode_encode_op, expected_value)
177 self.assertRaggedEqual(unicode_encode_op, expected_value)
[all …]
Dunicode_decode_op_test.py113 self.assertRaggedEqual(chars, expected_chars)
120 self.assertRaggedEqual(chars, expected_chars)
121 self.assertRaggedEqual(starts, [[0, 3, 6, 9], [0, 1, 2, 3, 4]])
137 self.assertRaggedEqual(expected, result)
155 self.assertRaggedEqual(expected_codepoints, result[0])
156 self.assertRaggedEqual(expected_offsets, result[1])
163 self.assertRaggedEqual(
165 self.assertRaggedEqual(
167 self.assertRaggedEqual(offsets, [[0, 1, 3, 5, 6, 7, 8, 9, 10], [0]])
266 self.assertRaggedEqual(expected, result)
[all …]

12