| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_split_op_test.py | 84 self.assertAllEqual(len(sp_tensors), 2)
85 self.assertAllEqual(
88 self.assertAllEqual(sp_tensors[0].values, [0, 2, 4, 5, 11, 13, 14])
89 self.assertAllEqual(sp_tensors[0].dense_shape, [2, 6])
90 self.assertAllEqual(
93 self.assertAllEqual(sp_tensors[1].values, [20, 23, 25, 30, 32, 33, 35])
94 self.assertAllEqual(sp_tensors[1].dense_shape, [2, 6])
101 self.assertAllEqual(len(sp_tensors_3), 3)
102 self.assertAllEqual(
105 self.assertAllEqual(sp_tensors_3[0].values, [0, 2, 11, 20, 30, 32, 41])
[all …]
|
| D | sparse_slice_op_test.py | 89 self.assertAllEqual(
92 self.assertAllEqual(sp_tensor0.values, [0, 2, 4, 5, 11, 13, 14])
93 self.assertAllEqual(sp_tensor0.dense_shape, [2, 6])
94 self.assertAllEqual(
97 self.assertAllEqual(sp_tensor1.values, [20, 23, 25, 30, 32, 33, 35])
98 self.assertAllEqual(sp_tensor1.dense_shape, [2, 6])
108 self.assertAllEqual(
111 self.assertAllEqual(sp_tensor0.values, [0, 2, 11, 20, 30, 32, 41])
112 self.assertAllEqual(sp_tensor0.dense_shape, [5, 3])
113 self.assertAllEqual(sp_tensor1.indices,
[all …]
|
| D | bcast_ops_test.py | 38 self.assertAllEqual(r, [2, 3, 5])
41 self.assertAllEqual(r, [2, 3, 5])
44 self.assertAllEqual(r, [2, 3, 5])
47 self.assertAllEqual(r, [2, 3, 5])
50 self.assertAllEqual(r, [2, 3, 5])
53 self.assertAllEqual(r, [2, 3, 5])
56 self.assertAllEqual(r, [2, 3, 5])
59 self.assertAllEqual(r, [2, 3, 5])
62 self.assertAllEqual(r, [2, 3, 5])
65 self.assertAllEqual(r, [2, 3, 5])
[all …]
|
| D | sparse_reshape_op_test.py | 59 self.assertAllEqual((5, 6), sp_input.get_shape())
61 self.assertAllEqual((1, 5, 2, 3), sp_output.get_shape())
66 self.assertAllEqual((2, 3, 4), sp_input.get_shape())
68 self.assertAllEqual((2, 3 * 4), sp_output.get_shape())
88 self.assertAllEqual((2, 3, 4), sp_input.shape)
94 self.assertAllEqual((2, 3 * 4), sp_output.shape)
102 self.assertAllEqual(output_val.indices, input_val.indices)
103 self.assertAllEqual(output_val.values, input_val.values)
104 self.assertAllEqual(output_val.dense_shape, input_val.dense_shape)
114 self.assertAllEqual(output_val.indices, input_val.indices)
[all …]
|
| D | lookup_ops_test.py | 90 self.assertAllEqual(3, self.evaluate(table.size()))
94 self.assertAllEqual([3], output.get_shape())
97 self.assertAllEqual([0, 1, -1], result)
113 self.assertAllEqual(3, self.evaluate(table.size()))
120 self.assertAllEqual([[0, 1], [-1, -1]], result)
131 self.assertAllEqual(3, self.evaluate(table.size()))
137 self.assertAllEqual([0, 1, -1], result)
147 self.assertAllEqual(3, self.evaluate(table.size()))
153 self.assertAllEqual([0, 1, -1], result)
170 self.assertAllEqual(3, self.evaluate(table1.size()))
[all …]
|
| D | sparse_serialization_ops_test.py | 79 self.assertAllEqual(indices, sp_input[0])
80 self.assertAllEqual(values, sp_input[1])
81 self.assertAllEqual(shape, sp_input[2])
106 self.assertAllEqual(combined_indices[:6, 0], [0] * 6) # minibatch 0
107 self.assertAllEqual(combined_indices[:6, 1:], sp_input[0])
108 self.assertAllEqual(combined_indices[6:, 0], [1] * 6) # minibatch 1
109 self.assertAllEqual(combined_indices[6:, 1:], sp_input[0])
110 self.assertAllEqual(combined_values[:6], sp_input[1])
111 self.assertAllEqual(combined_values[6:], sp_input[1])
112 self.assertAllEqual(combined_shape, [2, 5, 6])
[all …]
|
| D | string_split_op_test.py | 45 self.assertAllEqual(indices, [[0, 0], [0, 1], [0, 2], [0, 3], [1, 0]])
46 self.assertAllEqual(values, [b"pigs", b"on", b"the", b"wing", b"animals"])
47 self.assertAllEqual(shape, [2, 4])
56 self.assertAllEqual(indices, [[0, 0], [0, 1], [0, 2], [0, 3], [0, 4],
65 self.assertAllEqual(values.tolist(), expected)
66 self.assertAllEqual(shape, [3, 5])
74 self.assertAllEqual(
77 self.assertAllEqual(values, [b"a", b"b", b"c", b"d", b"e", b"f", b"g"])
78 self.assertAllEqual(shape, [10, 1])
86 self.assertAllEqual(
[all …]
|
| D | dynamic_partition_op_test.py | 50 self.assertAllEqual([0, 13], partition_vals[0])
51 self.assertAllEqual([17], partition_vals[1])
52 self.assertAllEqual([2, 4], partition_vals[2])
53 self.assertAllEqual([39], partition_vals[3])
73 self.assertAllEqual([[0, 1, 2], [3, 4, 5]], partition_vals[0])
74 self.assertAllEqual([[15, 16, 17]], partition_vals[1])
75 self.assertAllEqual([[6, 7, 8], [12, 13, 14]], partition_vals[2])
76 self.assertAllEqual([[9, 10, 11]], partition_vals[3])
98 self.assertAllEqual(part1, partition_vals[0])
99 self.assertAllEqual(part2, partition_vals[1])
[all …]
|
| D | sparse_reorder_op_test.py | 55 self.assertAllEqual((5, 6), sp_input.get_shape())
57 self.assertAllEqual((5, 6), sp_output.get_shape())
65 self.assertAllEqual(output_val.indices, input_val.indices)
66 self.assertAllEqual(output_val.values, input_val.values)
67 self.assertAllEqual(output_val.dense_shape, input_val.dense_shape)
77 self.assertAllEqual(output_val.indices, input_val.indices)
78 self.assertAllEqual(output_val.values, input_val.values)
79 self.assertAllEqual(output_val.dense_shape, input_val.dense_shape)
89 self.assertAllEqual(output_val.indices, expected_output_val.indices)
90 self.assertAllEqual(output_val.values, expected_output_val.values)
[all …]
|
| D | unicode_transcode_op_test.py | 47 self.assertAllEqual(values, strings)
57 self.assertAllEqual(values, strings)
67 self.assertAllEqual(values, strings)
82 self.assertAllEqual(values, expected)
95 self.assertAllEqual(values, b" ")
105 self.assertAllEqual(values, b"\x00 ")
118 self.assertAllEqual(values, b"abc abcdefg")
126 self.assertAllEqual(values, b"\x00\xef\xbf\xbd")
135 self.assertAllEqual(values, b"\x00 ")
172 self.assertAllEqual(values, b"\x00")
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_operators_test.py | 38 self.assertAllEqual(a == b, [[False, False], [True]])
39 self.assertAllEqual(a != b, [[True, True], [False]])
42 self.assertAllEqual(a == c, [[False, True], [False]])
43 self.assertAllEqual(a != c, [[True, False], [True]])
46 self.assertAllEqual(a == b, False)
47 self.assertAllEqual(a != b, True)
52 self.assertAllEqual((x > y), [[False, False], [True]])
53 self.assertAllEqual((x >= y), [[False, True], [True]])
54 self.assertAllEqual((x < y), [[True, False], [False]])
55 self.assertAllEqual((x <= y), [[True, True], [False]])
[all …]
|
| D | ragged_range_op_test.py | 33 self.assertAllEqual(rt1, [[0, 1, 2], [0, 1, 2, 3, 4], [0, 1]])
36 self.assertAllEqual(rt2, [[0, 1, 2], [], [8, 9, 10, 11]])
39 self.assertAllEqual(rt3, [[0, 2], [], [8, 10]])
43 self.assertAllEqual(
49 self.assertAllEqual(
56 self.assertAllEqual(
70 self.assertAllEqual(
76 self.assertAllEqual(
84 self.assertAllEqual(
91 self.assertAllEqual(
[all …]
|
| D | ragged_tensor_test.py | 72 self.assertAllEqual(rt, [[3, 1, 4, 1], [], [5, 9, 2], [6], []])
84 self.assertAllEqual(rt, [[3, 1, 4, 1], [], [5, 9, 2], [6], []])
93 self.assertAllEqual(outer_rt,
101 self.assertAllEqual(rt, [[[3, 1, 4, 1], [], [5, 9, 2]], [], [[6], []]])
107 self.assertAllEqual(
126 self.assertAllEqual(splits, rt_value.row_splits)
127 self.assertAllEqual(values, rt_value.values)
128 self.assertAllEqual(splits, rt_value.nested_row_splits[0])
129 self.assertAllEqual(values, rt_value.flat_values)
138 self.assertAllEqual(splits2, rt_value.row_splits)
[all …]
|
| D | ragged_to_sparse_op_test.py | 40 self.assertAllEqual(st.indices,
42 self.assertAllEqual(st.values, [1, 2, 3, 4, 5, 6])
43 self.assertAllEqual(st.dense_shape, [4, 3])
49 self.assertAllEqual(
51 self.assertAllEqual(st.values, b'a b c d e f g'.split())
52 self.assertAllEqual(st.dense_shape, [5, 3])
60 self.assertAllEqual(st.indices,
64 self.assertAllEqual(st.values,
66 self.assertAllEqual(st.dense_shape, [5, 3, 2])
73 self.assertAllEqual(st.values, [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12])
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable_test.py | 74 self.assertAllEqual(got, [2, 1])
80 self.assertAllEqual(got, [2, 1])
85 self.assertAllEqual(got, [6, 1])
90 self.assertAllEqual(got, [6, 1])
95 self.assertAllEqual(got, [2, 1])
99 self.assertAllEqual(got, [1, 1])
120 self.assertAllEqual(self.evaluate(s.variables[0]), [[4, 4]])
121 self.assertAllEqual(self.evaluate(s.variables[1]), [[5, 5], [6, 6]])
122 self.assertAllEqual(self.evaluate(s.variables[2]), [[7, 7]])
130 self.assertAllEqual(self.evaluate(s.variables[0]), [[1, 1]])
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 149 self.assertAllEqual([1, 2], node_ids)
171 self.assertAllEqual([1, 2], node_ids)
173 self.assertAllEqual([1, 1], thresholds)
174 self.assertAllEqual([1, 1], feature_dimensions)
179 self.assertAllEqual([_INEQUALITY_DEFAULT_LEFT] * 2, split_types)
199 self.assertAllEqual([[1, 2], [1, 2]], self.evaluate(node_ids_list))
202 self.assertAllEqual([[1, 1], [1, 1]], self.evaluate(thresholds_list))
252 self.assertAllEqual([1, 2], node_ids)
254 self.assertAllEqual([1, 1], thresholds)
255 self.assertAllEqual([12, 9], feature_ids)
[all …]
|
| /external/tensorflow/tensorflow/tools/compatibility/testdata/ |
| D | test_file_v0_11.py | 47 self.assertAllEqual(
50 self.assertAllEqual(
53 self.assertAllEqual(
56 self.assertAllEqual(
59 self.assertAllEqual(
62 self.assertAllEqual(tf.reduce_sum(a, [0, 1]).eval(), 21.0)
63 self.assertAllEqual(
66 self.assertAllEqual(
69 self.assertAllEqual(tf.reduce_prod(a, [0, 1]).eval(), 720.0)
70 self.assertAllEqual(
[all …]
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | convert_test.py | 51 self.assertAllEqual([1],
55 self.assertAllEqual([1], self.evaluate(
57 self.assertAllEqual([1], self.evaluate(
59 self.assertAllEqual([1],
66 self.assertAllEqual([-1],
70 self.assertAllEqual([-1],
72 self.assertAllEqual([-1],
74 self.assertAllEqual([-1],
76 self.assertAllEqual([-1],
94 self.assertAllEqual([3, 6],
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 52 self.assertAllEqual(15, product)
78 self.assertAllEqual(7, total)
84 self.assertAllEqual([[15]], product)
90 self.assertAllEqual([[0], [3]], x1)
91 self.assertAllEqual([[1], [4]], x2)
92 self.assertAllEqual([[2], [5]], x3)
106 self.assertAllEqual(r, 3)
114 self.assertAllEqual([[0, 1, 2], [3, 4, 5]], result[0])
129 self.assertAllEqual([2, 4, 6], out)
130 self.assertAllEqual([1, 3, 5], idx)
[all …]
|
| D | function_gradients_test.py | 75 self.assertAllEqual(step(), 2.0)
88 self.assertAllEqual(grads, 2.0)
240 self.assertAllEqual(self.evaluate(g).values, [[1.0]])
271 self.assertAllEqual(step_op(), 2.0)
284 self.assertAllEqual(self.evaluate(t.gradient(y, x)), 2.0)
299 self.assertAllEqual(self.evaluate(t.gradient(y, x)), 4.0)
326 self.assertAllEqual(backprop.implicit_grad(f)()[0][0], 2.0)
335 self.assertAllEqual(backprop.implicit_grad(f)()[0][0], 2.0)
337 self.assertAllEqual(backprop.implicit_grad(f)()[0][0], 2.0)
365 self.assertAllEqual(2, g(constant_op.constant(2.)))
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | normal_test.py | 61 self.assertAllEqual(all_true, is_finite)
66 self.assertAllEqual(expected, self.evaluate(mu_shape))
67 self.assertAllEqual(expected, self.evaluate(sigma_shape))
70 self.assertAllEqual(
106 self.assertAllEqual(self.evaluate(mu), self.evaluate(normal.loc))
107 self.assertAllEqual(
119 self.assertAllEqual(
121 self.assertAllEqual(
124 self.assertAllEqual(normal.batch_shape, log_pdf.get_shape())
125 self.assertAllEqual(normal.batch_shape, self.evaluate(log_pdf).shape)
[all …]
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_test.py | 92 self.assertAllEqual(result, [[42.0]])
101 self.assertAllEqual(result, [[42.0]])
111 self.assertAllEqual(arr, copy_val)
114 self.assertAllEqual(
125 self.assertAllEqual(inp, 10.0)
143 self.assertAllEqual(inp, 10.0)
151 self.assertAllEqual([10.0], results)
158 self.assertAllEqual([20.0], results)
169 self.assertAllEqual([30.0], results)
340 self.assertAllEqual(val1, result.field1)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | table_utils_test.py | 103 self.assertAllEqual(expected_indices, output_data.indices)
104 self.assertAllEqual(expected_values, output_data.values)
105 self.assertAllEqual(expected_dense_shape, output_data.dense_shape)
122 self.assertAllEqual(expected_indices, output_data.indices)
123 self.assertAllEqual(expected_values, output_data.values)
124 self.assertAllEqual(expected_dense_shape, output_data.dense_shape)
136 self.assertAllEqual(expected_output, output_data)
148 self.assertAllEqual(expected_output, output_data)
178 self.assertAllEqual(expected_indices, output_data.indices)
179 self.assertAllEqual(expected_values, output_data.values)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | ragged_keras_tensor_test.py | 57 self.assertAllEqual(inp.shape, [batch_size] + list(shape))
61 self.assertAllEqual(placeholder.shape, [batch_size] + list(shape))
69 self.assertAllEqual(model(x), x + x)
77 self.assertAllEqual(model(x), x * x)
85 self.assertAllEqual(model(x), x - x)
93 self.assertAllEqual(model(x), x / x)
108 self.assertAllEqual(model(x), expected_property)
113 self.assertAllEqual(model2(x), expected_property)
166 self.assertAllEqual(a, b)
173 self.assertAllEqual(a, b)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | bitwise_ops_test.py | 50 self.assertAllEqual(and_result, [0, 0, 3, 10])
51 self.assertAllEqual(or_result, [5, 5, 7, 15])
52 self.assertAllEqual(xor_result, [5, 5, 4, 5])
72 self.assertAllEqual(truth, popcnt_result)
91 self.assertAllEqual(not_a_and_a, [0, 0, 0, 0])
92 self.assertAllEqual(not_a_or_a, [not_0] * 4)
97 self.assertAllEqual(inverted, expected)
111 self.assertAllEqual(left_shift_result, np.left_shift(lhs, rhs))
112 self.assertAllEqual(right_shift_result, np.right_shift(lhs, rhs))
125 self.assertAllEqual(left_shift_result, np.left_shift(lhs, rhs))
[all …]
|