| /external/tensorflow/tensorflow/contrib/grid_rnn/python/kernel_tests/ |
| D | grid_rnn_test.py | 47 self.assertEqual(g[0].get_shape(), (1, 2))
48 self.assertEqual(s[0].c.get_shape(), (1, 2))
49 self.assertEqual(s[0].h.get_shape(), (1, 2))
50 self.assertEqual(s[1].c.get_shape(), (1, 2))
51 self.assertEqual(s[1].h.get_shape(), (1, 2))
75 self.assertEqual(g2[0].get_shape(), (1, 2))
76 self.assertEqual(s2[0].c.get_shape(), (1, 2))
77 self.assertEqual(s2[0].h.get_shape(), (1, 2))
78 self.assertEqual(s2[1].c.get_shape(), (1, 2))
79 self.assertEqual(s2[1].h.get_shape(), (1, 2))
[all …]
|
| /external/tensorflow/tensorflow/contrib/image/python/ops/ |
| D | image_ops.py | 73 elif image_or_images.get_shape().ndims is None:
75 elif len(image_or_images.get_shape()) == 2:
77 elif len(image_or_images.get_shape()) == 3:
79 elif len(image_or_images.get_shape()) == 4:
92 if image_or_images.get_shape().ndims is None:
94 elif len(image_or_images.get_shape()) == 2:
96 elif len(image_or_images.get_shape()) == 3:
150 if len(angle_or_angles.get_shape()) == 0: # pylint: disable=g-explicit-length-test
152 elif len(angle_or_angles.get_shape()) == 1:
193 if translation_or_translations.get_shape().ndims is None:
[all …]
|
| /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/contrib/losses/python/losses/ |
| D | loss_ops.py | 62 start_index = max(0, weights.get_shape().ndims)
63 axis = list(range(start_index, losses.get_shape().ndims))
107 if losses.get_shape().ndims is None:
109 weights_shape = weights.get_shape()
147 if weights.get_shape().ndims == 0:
160 if weights.get_shape().ndims >= 1:
161 axis = list(range(1, weights.get_shape().ndims))
168 array_ops.shape(losses), [weights.get_shape().ndims], [-1])
267 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
312 logits.get_shape().assert_is_compatible_with(multi_class_labels.get_shape())
[all …]
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | basic_decoder_test.py | 89 step_outputs[0].get_shape())
90 self.assertEqual((batch_size,), step_outputs[1].get_shape())
91 self.assertEqual((batch_size, cell_depth), first_state[0].get_shape())
92 self.assertEqual((batch_size, cell_depth), first_state[1].get_shape())
93 self.assertEqual((batch_size, cell_depth), step_state[0].get_shape())
94 self.assertEqual((batch_size, cell_depth), step_state[1].get_shape())
167 self.assertEqual((batch_size, cell_depth), step_outputs[0].get_shape())
168 self.assertEqual((batch_size,), step_outputs[1].get_shape())
169 self.assertEqual((batch_size, cell_depth), first_state[0].get_shape())
170 self.assertEqual((batch_size, cell_depth), first_state[1].get_shape())
[all …]
|
| D | basic_decoder_v2_test.py | 97 step_outputs[0].get_shape())
98 self.assertEqual((batch_size,), step_outputs[1].get_shape())
99 self.assertEqual((batch_size, cell_depth), first_state[0].get_shape())
100 self.assertEqual((batch_size, cell_depth), first_state[1].get_shape())
101 self.assertEqual((batch_size, cell_depth), step_state[0].get_shape())
102 self.assertEqual((batch_size, cell_depth), step_state[1].get_shape())
173 self.assertEqual((batch_size, cell_depth), step_outputs[0].get_shape())
174 self.assertEqual((batch_size,), step_outputs[1].get_shape())
175 self.assertEqual((batch_size, cell_depth), first_state[0].get_shape())
176 self.assertEqual((batch_size, cell_depth), first_state[1].get_shape())
[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 …]
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| D | geometric_test.py | 68 self.assertEqual([6,], log_prob.get_shape())
72 self.assertEqual([6,], pmf.get_shape())
93 self.assertEqual([6,], log_prob.get_shape())
95 self.assertEqual([6,], pmf.get_shape())
107 self.assertEqual([6, 3], log_prob.get_shape())
112 self.assertEqual([6, 3], pmf.get_shape())
126 self.assertEqual([6, 3], cdf.get_shape())
134 self.assertEqual([3], geom.entropy().get_shape())
142 self.assertEqual([3], geom.mean().get_shape())
150 self.assertEqual([3], geom.variance().get_shape())
[all …]
|
| D | binomial_test.py | 55 self.assertEqual((2, 1), binom.total_count.get_shape())
62 self.assertEqual((1, 3), binom.probs.get_shape())
63 self.assertEqual((1, 3), binom.logits.get_shape())
70 self.assertEqual((1, 3), binom.probs.get_shape())
71 self.assertEqual((1, 3), binom.logits.get_shape())
128 self.assertEqual((), pmf.get_shape())
129 self.assertEqual((), cdf.get_shape())
141 self.assertEqual((), pmf.get_shape())
142 self.assertEqual((), cdf.get_shape())
153 self.assertEqual((1, 2), pmf.get_shape())
[all …]
|
| D | poisson_test.py | 63 self.assertEqual(log_pmf.get_shape(), (6,))
67 self.assertEqual(pmf.get_shape(), (6,))
89 self.assertEqual(log_pmf.get_shape(), (6,))
91 self.assertEqual(pmf.get_shape(), (6,))
102 self.assertEqual(log_pmf.get_shape(), (6, 3))
106 self.assertEqual(pmf.get_shape(), (6, 3))
118 self.assertEqual(log_cdf.get_shape(), (6,))
122 self.assertEqual(cdf.get_shape(), (6,))
134 self.assertEqual(cdf.get_shape(), (6,))
153 self.assertEqual(log_cdf.get_shape(), (6, 3))
[all …]
|
| D | deterministic_test.py | 70 self.assertAllEqual((2, 2), prob.get_shape())
80 self.assertAllEqual((2, 2), prob.get_shape())
90 self.assertAllEqual((2, 2), prob.get_shape())
100 self.assertAllEqual((2, 2), prob.get_shape())
112 self.assertAllEqual((2, 3), prob.get_shape())
122 self.assertAllEqual((2, 2), cdf.get_shape())
132 self.assertAllEqual((2, 2), cdf.get_shape())
142 self.assertAllEqual((2, 2), cdf.get_shape())
150 self.assertAllEqual(sample_shape, sample.get_shape())
159 self.assertAllEqual(sample_shape + (2,), sample.get_shape())
[all …]
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/nets/ |
| D | vgg_test.py | 41 self.assertListEqual(logits.get_shape().as_list(),
52 self.assertListEqual(logits.get_shape().as_list(),
113 self.assertListEqual(logits.get_shape().as_list(),
116 self.assertListEqual(predictions.get_shape().as_list(), [batch_size])
128 self.assertListEqual(logits.get_shape().as_list(),
135 self.assertListEqual(logits.get_shape().as_list(),
139 self.assertEquals(predictions.get_shape().as_list(), [eval_batch_size])
162 self.assertListEqual(logits.get_shape().as_list(),
173 self.assertListEqual(logits.get_shape().as_list(),
247 self.assertListEqual(logits.get_shape().as_list(),
[all …]
|
| D | inception_v3_test.py | 46 self.assertListEqual(logits.get_shape().as_list(),
49 self.assertListEqual(end_points['Predictions'].get_shape().as_list(),
59 self.assertListEqual(final_endpoint.get_shape().as_list(),
119 self.assertListEqual(end_points[endpoint_name].get_shape().as_list(),
141 self.assertListEqual(logits.get_shape().as_list(),
145 self.assertListEqual(aux_logits.get_shape().as_list(),
149 self.assertListEqual(pre_pool.get_shape().as_list(),
153 self.assertListEqual(pre_logits.get_shape().as_list(),
173 original_depth = end_points[key].get_shape().as_list()[3]
174 new_depth = end_points_with_multiplier[key].get_shape().as_list()[3]
[all …]
|
| /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 …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 66 rank = ops.convert_to_tensor(x).get_shape().ndims
131 if np == 1 and (not transform_fn or ids.get_shape().ndims == 1):
159 params[0].get_shape()[0]))
162 params[p].get_shape()[0]))
168 param_p_dim = tensor_shape.dimension_value(params[p].get_shape()[0])
217 element_shape_s = params[0].get_shape()[1:]
219 element_shape_s = element_shape_s.merge_with(p.get_shape()[1:])
221 element_shape_s = ret.get_shape()[1:]
243 ret.set_shape(ids.get_shape().concatenate(element_shape_s))
467 sp_ids.values.get_shape().assert_is_compatible_with(
[all …]
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 128 sparse_ids.dense_shape.get_shape()[0]))
186 (original_rank_dim - 1).value).concatenate(result.get_shape()[1:]))
350 shape = params[p].get_shape()
498 embeds.set_shape(ids.get_shape().concatenate(
499 unique_embeddings.get_shape()[1:]))
649 sp_ids.values.get_shape().assert_is_compatible_with(
650 sp_weights.values.get_shape())
651 sp_ids.indices.get_shape().assert_is_compatible_with(
652 sp_weights.indices.get_shape())
653 sp_ids.dense_shape.get_shape().assert_is_compatible_with(
[all …]
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | sampling_ops.py | 206 target_probs.get_shape().assert_is_fully_defined()
208 labels, target_probs.get_shape().num_elements())
304 shape = tensor.get_shape().with_rank_at_least(1)
308 tensor_list[0].get_shape()[0])
319 probs.get_shape().assert_is_fully_defined()
322 probs.get_shape().assert_has_rank(1)
335 prob_length = checked_probs_list[0].get_shape().num_elements()
337 if checked_prob.get_shape().num_elements() != prob_length:
341 labels.get_shape().assert_has_rank(1)
345 shape = tensor.get_shape().with_rank_at_least(1)
[all …]
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/ |
| D | cholesky_outer_product_test.py | 49 self.assertAllEqual((2, 2, 2), bijector.forward(x).get_shape())
50 self.assertAllEqual((2, 2, 2), bijector.inverse(y).get_shape())
93 self.assertAllEqual([2, 2], y_actual.get_shape())
94 self.assertAllEqual([2, 2], x_actual.get_shape())
108 self.assertEqual(None, y_actual.get_shape())
109 self.assertEqual(None, x_actual.get_shape())
126 self.assertEqual([2, 2, 2], y_actual.get_shape())
127 self.assertEqual([2, 2, 2], x_actual.get_shape())
147 self.assertEqual(None, y_actual.get_shape())
148 self.assertEqual(None, x_actual.get_shape())
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | losses_impl.py | 258 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
312 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
360 logits.get_shape().assert_is_compatible_with(labels.get_shape())
429 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
495 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
566 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
648 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
705 logits.get_shape().assert_is_compatible_with(multi_class_labels.get_shape())
771 logits.get_shape().assert_is_compatible_with(onehot_labels.get_shape())
817 labels_rank = labels.get_shape().ndims
[all …]
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | pruning_utils.py | 50 var.get_shape(),
89 m1, n1 = mat1.get_shape().as_list()
91 m2, n2 = mat2.get_shape().as_list()
124 if tensor.get_shape().ndims != 2:
186 if input_tensor.get_shape().ndims != 2:
189 [height, width] = input_tensor.get_shape()
216 concatenated_variable_shape = partitioned_variable.get_shape()
218 partition_shape = partition.get_shape()
249 [partition.get_shape()[axis] for partition in partitioned_var])
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | adadelta_test.py | 79 self.assertEquals(slot[0].get_shape(), var0.get_shape())
83 self.assertEquals(slot_update[0].get_shape(), var0.get_shape())
87 self.assertEquals(slot[1].get_shape(), var1.get_shape())
91 self.assertEquals(slot_update[1].get_shape(), var1.get_shape())
|