| /external/tensorflow/tensorflow/python/layers/ |
| D | convolutional_test.py | 70 self.assertListEqual(output.get_shape().as_list(),
72 self.assertListEqual(layer.kernel.get_shape().as_list(), [3, 3, 4, 32])
73 self.assertListEqual(layer.bias.get_shape().as_list(), [32])
79 self.assertListEqual(output.get_shape().as_list(),
87 self.assertListEqual(output.get_shape().as_list(),
89 self.assertListEqual(layer.kernel.get_shape().as_list(), [3, 3, 4, 32])
90 self.assertListEqual(layer.bias.get_shape().as_list(), [32])
98 self.assertListEqual(output.get_shape().as_list(),
100 self.assertListEqual(layer.kernel.get_shape().as_list(), [3, 3, 4, 32])
101 self.assertListEqual(layer.bias.get_shape().as_list(), [32])
[all …]
|
| D | pooling_test.py | 59 self.assertListEqual(output.get_shape().as_list(), [5, 3, 4, 4])
66 self.assertListEqual(output.get_shape().as_list(), [5, 3, 4, 4])
76 self.assertListEqual(output.get_shape().as_list(), [5, 2, 6, 8])
87 self.assertListEqual(output.get_shape().as_list(), [3, 4, 4, 5])
99 self.assertListEqual(output.get_shape().as_list(), [None, 4, 4, 5])
107 self.assertListEqual(output.get_shape().as_list(),
116 self.assertListEqual(output.get_shape().as_list(),
126 self.assertListEqual(output.get_shape().as_list(),
136 self.assertListEqual(output.get_shape().as_list(),
144 self.assertListEqual(output.get_shape().as_list(), [5, 3, 3, 4, 4])
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | variable_ops_test.py | 70 self.assertEqual([1, 2], p.get_shape())
72 self.assertEqual(tensor_shape.unknown_shape(), p.get_shape())
94 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
102 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
108 self.assertEqual(tensor_shape.unknown_shape(), tensor.get_shape())
117 self.assertEqual(shape, var.get_shape())
118 self.assertEqual(shape, assigned.get_shape())
125 self.assertEqual(shape, var.get_shape())
127 self.assertEqual(tensor_shape.unknown_shape(), assigned.get_shape())
134 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
[all …]
|
| D | sparse_concat_op_test.py | 147 self.assertEqual(sp_concat.indices.get_shape(), [4, 2])
148 self.assertEqual(sp_concat.values.get_shape(), [4])
149 self.assertEqual(sp_concat.dense_shape.get_shape(), [2])
169 self.assertEqual(sp_concat.indices.get_shape(), [8, 2])
170 self.assertEqual(sp_concat.values.get_shape(), [8])
171 self.assertEqual(sp_concat.dense_shape.get_shape(), [2])
195 self.assertEqual(sp_concat.indices.get_shape(), [7, 2])
196 self.assertEqual(sp_concat.values.get_shape(), [7])
197 self.assertEqual(sp_concat.dense_shape.get_shape(), [2])
220 self.assertEqual(sp_concat.indices.get_shape(), [10, 2])
[all …]
|
| D | diag_op_test.py | 381 self.assertEqual((3, 3), v_diag.get_shape())
388 self.assertEqual(mat.shape, v_diag.get_shape())
396 self.assertEqual(v_diags.get_shape(), solution[0].shape)
406 self.assertEqual((2, 3, 3), v_batch_diag.get_shape())
416 self.assertEqual(mat_batch.shape, v_batch_diag.get_shape())
431 self.assertEqual(v_diags.get_shape(), solution.shape)
497 self.assertEqual(v_diags.get_shape(), solution.shape)
516 self.assertEqual(v_diags.get_shape(), solution.shape)
535 self.assertEqual(v_diags.get_shape(), solution.shape)
559 x.get_shape().as_list(),
[all …]
|
| D | candidate_sampler_ops_test.py | 64 self.assertEqual(sampled_candidates.get_shape(), [self.NUM_SAMPLED])
76 self.assertEqual(true_expected_count.get_shape(),
78 self.assertEqual(true_log_expected_count.get_shape(),
91 self.assertEqual(sampled_expected_count.get_shape(), [self.NUM_SAMPLED])
92 self.assertEqual(sampled_log_expected_count.get_shape(), [self.NUM_SAMPLED])
104 self.assertEqual(1, accidental_hits[0].get_shape().ndims)
105 self.assertEqual(1, accidental_hits[1].get_shape().ndims)
106 self.assertEqual(1, accidental_hits[2].get_shape().ndims)
|
| D | broadcast_to_ops_test.py | 126 shape = v_tf.get_shape().as_list()
146 err = gradient_checker.compute_gradient_error(x, x.get_shape(), out,
147 out.get_shape())
157 err = gradient_checker.compute_gradient_error(x, x.get_shape(),
158 out, out.get_shape())
168 err = gradient_checker.compute_gradient_error(x, x.get_shape(),
169 out, out.get_shape())
178 err = gradient_checker.compute_gradient_error(x, x.get_shape(),
179 out, out.get_shape())
191 err = gradient_checker.compute_gradient_error(x, x.get_shape(),
[all …]
|
| D | identity_op_py_test.py | 60 self.assertEqual(shape, tensor.get_shape())
61 self.assertEqual(shape, array_ops.identity(tensor).get_shape())
62 self.assertEqual(shape, array_ops.identity(array_2x3).get_shape())
64 array_ops.identity(np.array(array_2x3)).get_shape())
73 self.assertEqual(shape, tensor.get_shape())
74 self.assertEqual(shape, gen_array_ops.ref_identity(tensor).get_shape())
|
| D | constant_op_eager_test.py | 176 self.assertEqual(c.get_shape(), [2, 3, 5])
181 self.assertEqual(c.get_shape(), [2, 3, 5])
185 self.assertEqual(c.get_shape(), [7])
189 self.assertEqual(c.get_shape(), [7])
196 self.assertEqual(c.get_shape(), [5, 2, 3])
200 self.assertEqual(c.get_shape(), [7])
204 self.assertEqual(c.get_shape(), [1])
208 self.assertEqual(c.get_shape(), [])
223 self._testAll(constant_op.constant([1]).get_shape())
331 self.assertEqual(shape, ret.get_shape())
[all …]
|
| D | dynamic_stitch_op_test.py | 48 self.assertEqual([2], stitched_t.get_shape().as_list())
63 self.assertEqual([None], stitched_t.get_shape().as_list())
85 self.assertEqual([8], stitched_t.get_shape().as_list())
94 self.assertEqual([8], stitched_t.get_shape().as_list())
112 self.assertEqual([8, 2], stitched_t.get_shape().as_list())
132 self.assertEqual([8, 2], stitched_t.get_shape().as_list())
146 self.assertEqual([0, 2], stitched_t.get_shape().as_list())
165 self.assertEqual([7, 2], stitched_t.get_shape().as_list())
249 self.assertEqual([2], stitched_t.get_shape().as_list())
268 self.assertEqual([7, 2], stitched_t.get_shape().as_list())
[all …]
|
| D | map_fn_test.py | 147 self.assertEqual((6,), r[0].get_shape())
148 self.assertEqual((6,), r[1].get_shape())
170 self.assertEqual((6,), r.get_shape())
180 self.assertEqual((6,), r[0].get_shape())
181 self.assertEqual((6,), r[1].get_shape())
182 self.assertEqual((6,), r[2].get_shape())
192 self.assertAllEqual(y.get_shape(), self.evaluate(y).shape)
198 self.assertIs(None, y.get_shape().dims)
206 self.assertAllEqual([0], map_return.get_shape().dims)
216 self.assertAllEqual([0, 3, 2], map_return.get_shape().dims)
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | multinomial_test.py | 64 self.assertEqual((2, 1), dist.total_count.get_shape())
72 self.assertEqual((1, 3), dist.probs.get_shape())
73 self.assertEqual((1, 3), dist.logits.get_shape())
82 self.assertEqual((1, 3), multinom.probs.get_shape())
83 self.assertEqual((1, 3), multinom.logits.get_shape())
140 self.assertEqual((), pmf.get_shape())
151 self.assertEqual((), pmf.get_shape())
159 self.assertEqual((2), pmf.get_shape())
167 self.assertEqual((2), pmf.get_shape())
176 self.assertEqual((2), pmf.get_shape())
[all …]
|
| D | normal_test.py | 120 self.evaluate(normal.batch_shape_tensor()), log_pdf.get_shape())
124 self.assertAllEqual(normal.batch_shape, log_pdf.get_shape())
129 self.evaluate(normal.batch_shape_tensor()), pdf.get_shape())
133 self.assertAllEqual(normal.batch_shape, pdf.get_shape())
154 self.assertEqual(log_pdf.get_shape(), (6, 2))
156 self.evaluate(normal.batch_shape_tensor()), log_pdf.get_shape())
160 self.assertAllEqual(normal.batch_shape, log_pdf.get_shape())
165 self.assertEqual(pdf.get_shape(), (6, 2))
167 self.evaluate(normal.batch_shape_tensor()), pdf.get_shape())
170 self.assertAllEqual(normal.batch_shape, pdf.get_shape())
[all …]
|
| D | dirichlet_multinomial_test.py | 67 self.assertEqual([1, 1], dist.total_count.get_shape())
75 self.assertEqual([1, 3], dist.concentration.get_shape())
121 self.assertEqual((), pmf.get_shape())
133 self.assertEqual((), pmf.get_shape())
145 self.assertEqual((4, 3), pmf.get_shape())
156 self.assertAllEqual([2], pmf.get_shape())
166 self.assertAllEqual([2], pmf.get_shape())
176 self.assertAllEqual([2], pmf.get_shape())
186 self.assertAllEqual([2], pmf.get_shape())
301 self.assertEqual([2, 2], covariance.get_shape())
[all …]
|
| D | laplace_test.py | 69 self.assertEqual(log_pdf.get_shape(), (6,))
76 self.assertEqual(pdf.get_shape(), (6,))
89 self.assertEqual(log_pdf.get_shape(), (6, 2))
93 self.assertEqual(pdf.get_shape(), (6, 2))
110 self.assertEqual(log_pdf.get_shape(), (6, 2))
114 self.assertEqual(pdf.get_shape(), (6, 2))
132 self.assertEqual(cdf.get_shape(), (6,))
149 self.assertEqual(cdf.get_shape(), (6,))
166 self.assertEqual(sf.get_shape(), (6,))
176 self.assertEqual(laplace.mean().get_shape(), (3,))
[all …]
|
| D | dirichlet_test.py | 70 self.assertEqual([1, 3], dist.concentration.get_shape())
109 self.assertEqual((), pdf.get_shape())
117 self.assertEqual((), pdf.get_shape())
126 self.assertEqual((2), pdf.get_shape())
134 self.assertEqual((2), pdf.get_shape())
141 self.assertEqual((2), pdf.get_shape())
148 self.assertEqual((2), pdf.get_shape())
155 self.assertEqual((2), pdf.get_shape())
160 self.assertEqual(dirichlet.mean().get_shape(), [3])
207 self.assertEqual(dirichlet.covariance().get_shape(), (3, 3))
[all …]
|
| D | gamma_test.py | 71 self.assertEqual(log_pdf.get_shape(), (6,))
73 self.assertEqual(pdf.get_shape(), (6,))
98 self.assertEqual(log_pdf.get_shape(), (6, 2))
101 self.assertEqual(pdf.get_shape(), (6, 2))
118 self.assertEqual(log_pdf.get_shape(), (6, 2))
121 self.assertEqual(pdf.get_shape(), (6, 2))
139 self.assertEqual(cdf.get_shape(), (6,))
149 self.assertEqual(gamma.mean().get_shape(), (3,))
160 self.assertEqual(gamma.mode().get_shape(), (3,))
180 self.assertEqual(gamma.mode().get_shape(), (3,))
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 68 rank = ops.convert_to_tensor(x).get_shape().ndims
134 if np == 1 and (not transform_fn or ids.get_shape().ndims == 1):
162 tensor_shape.dimension_value(params[0].get_shape()[0]))
165 tensor_shape.dimension_value(params[p].get_shape()[0]))
171 param_p_dim = tensor_shape.dimension_value(params[p].get_shape()[0])
220 element_shape_s = params[0].get_shape()[1:]
222 element_shape_s = element_shape_s.merge_with(p.get_shape()[1:])
224 element_shape_s = ret.get_shape()[1:]
245 ret.set_shape(ids.get_shape().concatenate(element_shape_s))
469 sp_ids.values.get_shape().assert_is_compatible_with(
[all …]
|
| D | bitwise_ops_test.py | 164 self.assertAllEqual(and_tensor.get_shape().as_list(), and_result.shape)
165 self.assertAllEqual(and_tensor.get_shape().as_list(), [3, 3])
166 self.assertAllEqual(or_tensor.get_shape().as_list(), or_result.shape)
167 self.assertAllEqual(or_tensor.get_shape().as_list(), [3, 3])
168 self.assertAllEqual(xor_tensor.get_shape().as_list(), xor_result.shape)
169 self.assertAllEqual(xor_tensor.get_shape().as_list(), [3, 3])
170 self.assertAllEqual(ls_tensor.get_shape().as_list(), ls_result.shape)
171 self.assertAllEqual(ls_tensor.get_shape().as_list(), [3, 3])
172 self.assertAllEqual(rs_tensor.get_shape().as_list(), rs_result.shape)
173 self.assertAllEqual(rs_tensor.get_shape().as_list(), [3, 3])
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | losses_impl.py | 253 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
307 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
355 logits.get_shape().assert_is_compatible_with(labels.get_shape())
422 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
488 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
559 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
641 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
698 logits.get_shape().assert_is_compatible_with(multi_class_labels.get_shape())
764 logits.get_shape().assert_is_compatible_with(onehot_labels.get_shape())
810 labels_rank = labels.get_shape().ndims
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | adadelta_test.py | 79 self.assertEqual(slot[0].get_shape(), var0.get_shape())
83 self.assertEqual(slot_update[0].get_shape(), var0.get_shape())
87 self.assertEqual(slot[1].get_shape(), var1.get_shape())
91 self.assertEqual(slot_update[1].get_shape(), var1.get_shape())
|
| D | momentum_test.py | 57 self.assertEquals(slot0.get_shape(), var0.get_shape())
60 self.assertEquals(slot1.get_shape(), var1.get_shape())
143 self.assertEquals(slot0.get_shape(), var0.get_shape())
146 self.assertEquals(slot1.get_shape(), var1.get_shape())
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | random_poisson_test.py | 118 self.assertEqual([0], rnd.get_shape().as_list())
125 self.assertEqual([150], rnd.get_shape().as_list())
130 self.assertEqual([150, 1, 2, 3], rnd.get_shape().as_list())
135 self.assertEqual([20, 30, 1, 2, 3], rnd.get_shape().as_list())
140 self.assertEqual([12, 2], rnd.get_shape().as_list())
146 self.assertEqual([None, 7, 3], rnd.get_shape().as_list())
151 self.assertEqual([None, None, None, 9, 6], rnd.get_shape().as_list())
157 self.assertIs(None, rnd.get_shape().ndims)
162 self.assertIs(None, rnd.get_shape().ndims)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | input_test.py | 800 self.assertAllEqual((1,), sparse.dense_shape.get_shape().as_list())
802 self.assertAllEqual((2,), batched.dense_shape.get_shape().as_list())
808 self.assertAllEqual((1,), sparse.dense_shape.get_shape().as_list())
810 self.assertAllEqual((1,), batched.dense_shape.get_shape().as_list())
818 self.assertIs(None, sparse.dense_shape.get_shape().num_elements())
820 self.assertIs(None, batched.dense_shape.get_shape().num_elements())
828 self.assertIs(None, sparse.dense_shape.get_shape().num_elements())
830 self.assertIs(None, batched.dense_shape.get_shape().num_elements())
835 self.assertAllEqual((8, 2), x["c"].get_shape().as_list())
926 self.assertAllEqual((1,), sparse.dense_shape.get_shape().as_list())
[all …]
|
| D | adadelta_test.py | 89 self.assertEquals(slot[0].get_shape(), var0.get_shape())
93 self.assertEquals(slot_update[0].get_shape(), var0.get_shape())
97 self.assertEquals(slot[1].get_shape(), var1.get_shape())
101 self.assertEquals(slot_update[1].get_shape(), var1.get_shape())
|