| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_split_op_test.py | 84 self.assertAllEqual(len(sp_tensors), 2)
85 self.assertAllEqual(sp_tensors[0].indices.eval(), [[0, 0], [0, 2], [0, 4],
88 self.assertAllEqual(sp_tensors[0].values.eval(), [0, 2, 4, 5, 11, 13, 14])
89 self.assertAllEqual(sp_tensors[0].dense_shape.eval(), [2, 6])
90 self.assertAllEqual(sp_tensors[1].indices.eval(), [[0, 0], [0, 3], [0, 5],
93 self.assertAllEqual(sp_tensors[1].values.eval(),
95 self.assertAllEqual(sp_tensors[1].dense_shape.eval(), [2, 6])
102 self.assertAllEqual(len(sp_tensors_3), 3)
103 self.assertAllEqual(sp_tensors_3[0].indices.eval(),
106 self.assertAllEqual(sp_tensors_3[0].values.eval(),
[all …]
|
| D | sparse_slice_op_test.py | 89 self.assertAllEqual(
92 self.assertAllEqual(sp_tensor0.values.eval(), [0, 2, 4, 5, 11, 13, 14])
93 self.assertAllEqual(sp_tensor0.dense_shape.eval(), [2, 6])
94 self.assertAllEqual(
97 self.assertAllEqual(sp_tensor1.values.eval(),
99 self.assertAllEqual(sp_tensor1.dense_shape.eval(), [2, 6])
109 self.assertAllEqual(
112 self.assertAllEqual(sp_tensor0.values.eval(), [0, 2, 11, 20, 30, 32, 41])
113 self.assertAllEqual(sp_tensor0.dense_shape.eval(), [5, 3])
114 self.assertAllEqual(sp_tensor1.indices.eval(),
[all …]
|
| D | bcast_ops_test.py | 42 self.assertAllEqual(r, [2, 3, 5])
45 self.assertAllEqual(r, [2, 3, 5])
48 self.assertAllEqual(r, [2, 3, 5])
51 self.assertAllEqual(r, [2, 3, 5])
54 self.assertAllEqual(r, [2, 3, 5])
57 self.assertAllEqual(r, [2, 3, 5])
60 self.assertAllEqual(r, [2, 3, 5])
63 self.assertAllEqual(r, [2, 3, 5])
66 self.assertAllEqual(r, [2, 3, 5])
69 self.assertAllEqual(r, [2, 3, 5])
[all …]
|
| D | string_split_op_test.py | 39 self.assertAllEqual(indices, [[0, 0], [0, 1], [0, 2], [0, 3], [1, 0]])
40 self.assertAllEqual(values, [b"pigs", b"on", b"the", b"wing", b"animals"])
41 self.assertAllEqual(shape, [2, 4])
50 self.assertAllEqual(indices, [[0, 0], [0, 1], [0, 2], [0, 3], [0, 4],
59 self.assertAllEqual(values.tolist(), expected)
60 self.assertAllEqual(shape, [3, 5])
68 self.assertAllEqual(
71 self.assertAllEqual(values, [b"a", b"b", b"c", b"d", b"e", b"f", b"g"])
72 self.assertAllEqual(shape, [10, 1])
80 self.assertAllEqual(
[all …]
|
| D | sparse_reshape_op_test.py | 57 self.assertAllEqual((5, 6), sp_input.get_shape())
59 self.assertAllEqual((1, 5, 2, 3), sp_output.get_shape())
64 self.assertAllEqual((2, 3, 4), sp_input.get_shape())
66 self.assertAllEqual((2, 3 * 4), sp_output.get_shape())
88 self.assertAllEqual(output_val.indices, input_val.indices)
89 self.assertAllEqual(output_val.values, input_val.values)
90 self.assertAllEqual(output_val.dense_shape, input_val.dense_shape)
100 self.assertAllEqual(output_val.indices, input_val.indices)
101 self.assertAllEqual(output_val.values, input_val.values)
102 self.assertAllEqual(output_val.dense_shape, input_val.dense_shape)
[all …]
|
| D | lookup_ops_test.py | 75 self.assertAllEqual(3, self.evaluate(table.size()))
79 self.assertAllEqual([3], output.get_shape())
82 self.assertAllEqual([0, 1, -1], result)
98 self.assertAllEqual(3, self.evaluate(table.size()))
105 self.assertAllEqual([[0, 1], [-1, -1]], result)
116 self.assertAllEqual(3, self.evaluate(table.size()))
122 self.assertAllEqual([0, 1, -1], result)
132 self.assertAllEqual(3, self.evaluate(table.size()))
138 self.assertAllEqual([0, 1, -1], result)
155 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 | dynamic_partition_op_test.py | 48 self.assertAllEqual([0, 13], partition_vals[0])
49 self.assertAllEqual([17], partition_vals[1])
50 self.assertAllEqual([2, 4], partition_vals[2])
51 self.assertAllEqual([39], partition_vals[3])
71 self.assertAllEqual([[0, 1, 2], [3, 4, 5]], partition_vals[0])
72 self.assertAllEqual([[15, 16, 17]], partition_vals[1])
73 self.assertAllEqual([[6, 7, 8], [12, 13, 14]], partition_vals[2])
74 self.assertAllEqual([[9, 10, 11]], partition_vals[3])
96 self.assertAllEqual(part1, partition_vals[0])
97 self.assertAllEqual(part2, partition_vals[1])
[all …]
|
| D | bincount_op_test.py | 35 self.assertAllEqual(self.evaluate(math_ops.bincount([], minlength=5)),
37 self.assertAllEqual(self.evaluate(math_ops.bincount([], minlength=1)),
39 self.assertAllEqual(self.evaluate(math_ops.bincount([], minlength=0)),
50 self.assertAllEqual(self.evaluate(math_ops.bincount([1, 1, 1, 2, 2, 3])),
53 self.assertAllEqual(self.evaluate(math_ops.bincount(arr)),
56 self.assertAllEqual(self.evaluate(math_ops.bincount(arr)),
59 self.assertAllEqual(self.evaluate(math_ops.bincount([])), [])
60 self.assertAllEqual(self.evaluate(math_ops.bincount([0, 0, 0])), [3])
61 self.assertAllEqual(self.evaluate(math_ops.bincount([5])),
63 self.assertAllEqual(self.evaluate(math_ops.bincount(np.arange(10000))),
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator_signals_test.py | 81 self.assertAllEqual(a[:batch_size], result['a'])
82 self.assertAllEqual(b[:batch_size], result['b'])
86 self.assertAllEqual(a[batch_size:num_samples], result['a'])
87 self.assertAllEqual(b[batch_size:num_samples], result['b'])
114 self.assertAllEqual(a[:batch_size], result['a'])
115 self.assertAllEqual(b[:batch_size], result['b'])
116 self.assertAllEqual([[0.]] * batch_size, evaluated_signals['stopping'])
120 self.assertAllEqual(a[batch_size:num_samples], result['a'])
121 self.assertAllEqual(b[batch_size:num_samples], result['b'])
122 self.assertAllEqual([[0.]] * batch_size, evaluated_signals['stopping'])
[all …]
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | drop_stale_gradient_optimizer_test.py | 127 self.assertAllEqual(0.0, sessions[0].run(var_0))
128 self.assertAllEqual(1.0, sessions[0].run(var_1))
129 self.assertAllEqual(0.0, sessions[0].run(stale_counter))
130 self.assertAllEqual(0, sessions[0].run(global_step))
135 self.assertAllEqual(1, sessions[0].run(global_step))
136 self.assertAllEqual(0.0 + 1.0, sessions[0].run(var_0))
137 self.assertAllEqual(1.0 + 1.0, sessions[0].run(var_1))
138 self.assertAllEqual(1, sessions[0].run(global_step))
150 self.assertAllEqual(0.0, sessions[0].run(var_0))
151 self.assertAllEqual(1.0, sessions[0].run(var_1))
[all …]
|
| D | elastic_average_optimizer_test.py | 166 self.assertAllEqual(0.0, sessions[0].run(var_0))
167 self.assertAllEqual(1.0, sessions[0].run(var_1))
168 self.assertAllEqual(0.0, sessions[0].run(var_0_g))
169 self.assertAllEqual(1.0, sessions[0].run(var_1_g))
170 self.assertAllEqual(0, sessions[0].run(global_step))
174 self.assertAllEqual(1.0, sessions[0].run(var_0))
175 self.assertAllEqual(2.0, sessions[0].run(var_1))
176 self.assertAllEqual(0.0, sessions[0].run(var_0_g))
177 self.assertAllEqual(1.0, sessions[0].run(var_1_g))
178 self.assertAllEqual(0, sessions[0].run(global_step))
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 49 self.assertAllEqual(15, product)
76 self.assertAllEqual(7, total)
82 self.assertAllEqual([[15]], product)
88 self.assertAllEqual([[0], [3]], x1)
89 self.assertAllEqual([[1], [4]], x2)
90 self.assertAllEqual([[2], [5]], x3)
105 self.assertAllEqual(r, 3)
113 self.assertAllEqual([[0, 1, 2], [3, 4, 5]], result[0])
128 self.assertAllEqual([2, 4, 6], out)
129 self.assertAllEqual([1, 3, 5], idx)
[all …]
|
| D | function_gradients_test.py | 55 self.assertAllEqual(step(), 2.0)
68 self.assertAllEqual(grads.eval(), 2.0)
81 self.assertAllEqual(self.evaluate(g).values, [[1.0]])
112 self.assertAllEqual(step_op(), 2.0)
125 self.assertAllEqual(self.evaluate(t.gradient(y, x)), 2.0)
140 self.assertAllEqual(self.evaluate(t.gradient(y, x)), 4.0)
149 self.assertAllEqual(backprop.implicit_grad(f)()[0][0], 2.0)
158 self.assertAllEqual(backprop.implicit_grad(f)()[0][0], 2.0)
160 self.assertAllEqual(backprop.implicit_grad(f)()[0][0], 2.0)
188 self.assertAllEqual(2, g(constant_op.constant(2.)))
[all …]
|
| /external/tensorflow/tensorflow/tools/compatibility/testdata/ |
| D | test_file_v0_11.py | 46 self.assertAllEqual(
49 self.assertAllEqual(
52 self.assertAllEqual(
55 self.assertAllEqual(
58 self.assertAllEqual(
61 self.assertAllEqual(tf.reduce_sum(a, [0, 1]).eval(), 21.0)
62 self.assertAllEqual(
65 self.assertAllEqual(
68 self.assertAllEqual(tf.reduce_prod(a, [0, 1]).eval(), 720.0)
69 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/contrib/distributions/python/kernel_tests/ |
| D | cauchy_test.py | 56 self.assertAllEqual(all_true, is_finite)
62 self.assertAllEqual(expected, loc_shape.eval())
63 self.assertAllEqual(expected, scale_shape.eval())
66 self.assertAllEqual(expected,
96 self.assertAllEqual(cauchy.batch_shape_tensor().eval(), log_pdf.shape)
97 self.assertAllEqual(cauchy.batch_shape_tensor().eval(),
99 self.assertAllEqual(cauchy.batch_shape, log_pdf.shape)
100 self.assertAllEqual(cauchy.batch_shape, log_pdf.eval().shape)
103 self.assertAllEqual(cauchy.batch_shape_tensor().eval(), pdf.shape)
104 self.assertAllEqual(cauchy.batch_shape_tensor().eval(), pdf.eval().shape)
[all …]
|
| D | deterministic_test.py | 36 self.assertAllEqual(deterministic.batch_shape_tensor().eval(), (2, 3, 4))
37 self.assertAllEqual(deterministic.batch_shape, (2, 3, 4))
38 self.assertAllEqual(deterministic.event_shape_tensor().eval(), [])
70 self.assertAllEqual((2, 2), prob.get_shape())
71 self.assertAllEqual(expected_prob, prob.eval())
80 self.assertAllEqual((2, 2), prob.get_shape())
81 self.assertAllEqual(expected_prob, prob.eval())
90 self.assertAllEqual((2, 2), prob.get_shape())
91 self.assertAllEqual(expected_prob, prob.eval())
100 self.assertAllEqual((2, 2), prob.get_shape())
[all …]
|
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| D | sparse_ops_test.py | 32 test_case.assertAllEqual(expected.indices, actual.indices)
36 test_case.assertAllEqual(expected.values, actual.values)
39 test_case.assertAllEqual(expected.dense_shape, actual.dense_shape)
51 self.assertAllEqual([[0], [2]], result.indices)
52 self.assertAllEqual([1, 2], result.values)
53 self.assertAllEqual([4], result.dense_shape)
62 self.assertAllEqual([[0], [2]], result.indices)
64 self.assertAllEqual([4], result.dense_shape)
73 self.assertAllEqual([[0], [2]], result.indices)
74 self.assertAllEqual([True, True], result.values)
[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 | 89 self.assertAllEqual(result, [[42.0]])
98 self.assertAllEqual(result, [[42.0]])
108 self.assertAllEqual(arr, copy_val)
111 self.assertAllEqual(
122 self.assertAllEqual(inp.eval(), 10.0)
140 self.assertAllEqual(inp.eval(), 10.0)
148 self.assertAllEqual([10.0], results)
155 self.assertAllEqual([20.0], results)
166 self.assertAllEqual([30.0], results)
337 self.assertAllEqual(val1, result.field1)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tensor_array_ops_test.py | 68 self.assertAllEqual([[4.0, 5.0]], d0)
69 self.assertAllEqual([[1.0, 3.0]], d1)
70 self.assertAllEqual([[7.0, -8.5]], d2)
71 self.assertAllEqual([], flow_val.shape)
86 self.assertAllEqual(
106 self.assertAllEqual([3, 0, 1], self.evaluate(c0).shape)
121 self.assertAllEqual(
144 self.assertAllEqual(convert(1.0), d0)
145 self.assertAllEqual(convert(2.0), d1)
146 self.assertAllEqual(convert(3.0), d2)
[all …]
|
| /external/tensorflow/tensorflow/contrib/timeseries/examples/ |
| D | known_anomaly_test.py | 32 self.assertAllEqual(
35 self.assertAllEqual(all_times.shape, mean.shape)
36 self.assertAllEqual(all_times.shape, upper_limit.shape)
37 self.assertAllEqual(all_times.shape, lower_limit.shape)
38 self.assertAllEqual(times.shape, observed.shape)
44 self.assertAllEqual(
47 self.assertAllEqual([200], times.shape)
48 self.assertAllEqual([200], observed.shape)
49 self.assertAllEqual([300], all_times.shape)
50 self.assertAllEqual([300], mean.shape)
[all …]
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sharded_mutable_dense_hashtable_test.py | 51 self.assertAllEqual(0, table.size().eval())
54 self.assertAllEqual(3, table.size().eval())
58 self.assertAllEqual([3], output.get_shape())
59 self.assertAllEqual([0, 1, -1], output.eval())
78 self.assertAllEqual(0, table.size().eval())
81 self.assertAllEqual(3, table.size().eval())
86 self.assertAllEqual([3, 2], output.get_shape())
105 self.assertAllEqual(0, table.size().eval())
108 self.assertAllEqual(3, table.size().eval())
111 self.assertAllEqual(num_shards, len(keys_list))
[all …]
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | iterator_checkpoint_test.py | 47 self.assertAllEqual([1, 4], get_next())
49 self.assertAllEqual([9, 16], get_next())
50 self.assertAllEqual([25, 36], get_next())
52 self.assertAllEqual([9, 16], get_next())
53 self.assertAllEqual([25, 36], get_next())
78 self.assertAllEqual([1, 4], get_next_1())
79 self.assertAllEqual(0, get_next_3())
80 self.assertAllEqual(1, get_next_3())
81 self.assertAllEqual(2, get_next_3())
83 self.assertAllEqual([1, 4], get_next_2())
[all …]
|