Home
last modified time | relevance | path

Searched refs:get_shape (Results 1 – 25 of 282) sorted by relevance

12345678910>>...12

/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional_test.py70 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 …]
Dpooling_test.py59 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/
Dvariable_ops_test.py76 self.assertEqual([1, 2], p.get_shape())
78 self.assertEqual(tensor_shape.unknown_shape(), p.get_shape())
100 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
108 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
114 self.assertEqual(tensor_shape.unknown_shape(), tensor.get_shape())
123 self.assertEqual(shape, var.get_shape())
124 self.assertEqual(shape, assigned.get_shape())
131 self.assertEqual(shape, var.get_shape())
133 self.assertEqual(tensor_shape.unknown_shape(), assigned.get_shape())
140 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
[all …]
Dsparse_concat_op_test.py147 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 …]
Ddiag_op_test.py381 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)
558 x.get_shape().as_list(),
[all …]
Dreshape_op_test.py40 self.assertEqual(tf_ans.get_shape(), out.shape)
47 self.assertEqual(tf_ans.get_shape(), out.shape)
168 self.assertEqual([1, 1, None, 1], y.get_shape().as_list())
172 self.assertEqual(None, y.get_shape().ndims)
176 self.assertEqual([None, None, None], y.get_shape().as_list())
180 self.assertEqual([None, 37], y.get_shape().as_list())
188 self.assertEqual([None, None, 37, 42], y.get_shape().as_list())
196 self.assertEqual([None, 37, None], y.get_shape().as_list())
203 self.assertEqual([100, 1], y.get_shape().as_list())
205 self.assertEqual([100, 1], y.get_shape().as_list())
[all …]
Dcandidate_sampler_ops_test.py64 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)
Ddynamic_stitch_op_test.py48 self.assertEqual([2], stitched_t.get_shape().as_list())
63 self.assertEqual([None], stitched_t.get_shape().as_list())
86 self.assertEqual([8], stitched_t.get_shape().as_list())
95 self.assertEqual([8], stitched_t.get_shape().as_list())
113 self.assertEqual([8, 2], stitched_t.get_shape().as_list())
133 self.assertEqual([8, 2], stitched_t.get_shape().as_list())
147 self.assertEqual([0, 2], stitched_t.get_shape().as_list())
166 self.assertEqual([7, 2], stitched_t.get_shape().as_list())
250 self.assertEqual([2], stitched_t.get_shape().as_list())
269 self.assertEqual([7, 2], stitched_t.get_shape().as_list())
[all …]
Dconstant_op_eager_test.py191 self.assertEqual(c.get_shape(), [2, 3, 5])
196 self.assertEqual(c.get_shape(), [2, 3, 5])
200 self.assertEqual(c.get_shape(), [7])
204 self.assertEqual(c.get_shape(), [7])
211 self.assertEqual(c.get_shape(), [5, 2, 3])
215 self.assertEqual(c.get_shape(), [7])
219 self.assertEqual(c.get_shape(), [1])
223 self.assertEqual(c.get_shape(), [])
238 self._testAll(constant_op.constant([1]).get_shape())
346 self.assertEqual(shape, ret.get_shape())
[all …]
Dconstant_op_test.py199 self.assertEqual(c.get_shape(), [2, 3, 5])
210 self.assertEqual(c.get_shape(), [2, 3, 5])
215 self.assertEqual(c.get_shape(), [7])
220 self.assertEqual(c.get_shape(), [7])
225 self.assertEqual(c.get_shape(), [1])
230 self.assertEqual(c.get_shape(), [])
235 self.assertEqual(c.get_shape(), [10])
244 self.assertEqual(c.get_shape(), [10])
247 self.assertEqual(c.get_shape(), [10])
399 self.assertEqual(shape, ret.get_shape())
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dmultinomial_test.py64 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 …]
Dnormal_test.py120 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 …]
Ddirichlet_multinomial_test.py67 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())
303 self.assertEqual([2, 2], covariance.get_shape())
[all …]
Dlaplace_test.py69 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 …]
Ddirichlet_test.py70 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])
209 self.assertEqual(dirichlet.covariance().get_shape(), (3, 3))
[all …]
Dgamma_test.py71 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 …]
Dexponential_test.py57 self.assertEqual(log_pdf.get_shape(), (6,))
60 self.assertEqual(pdf.get_shape(), (6,))
84 self.assertEqual(cdf.get_shape(), (6,))
100 self.assertEqual(log_survival.get_shape(), (7,))
110 self.assertEqual(exponential.mean().get_shape(), (3,))
119 self.assertEqual(exponential.variance().get_shape(), (3,))
129 self.assertEqual(exponential.entropy().get_shape(), (3,))
161 self.assertEqual(samples.get_shape(), (n, batch_size, 2))
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py69 rank = ops.convert_to_tensor(x).get_shape().ndims
135 if np == 1 and (not transform_fn or ids.get_shape().ndims == 1):
163 tensor_shape.dimension_value(params[0].get_shape()[0]))
166 tensor_shape.dimension_value(params[p].get_shape()[0]))
172 param_p_dim = tensor_shape.dimension_value(params[p].get_shape()[0])
221 element_shape_s = params[0].get_shape()[1:]
223 element_shape_s = element_shape_s.merge_with(p.get_shape()[1:])
225 element_shape_s = ret.get_shape()[1:]
246 ret.set_shape(ids.get_shape().concatenate(element_shape_s))
495 sp_ids.values.get_shape().assert_is_compatible_with(
[all …]
Dbitwise_ops_test.py164 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/
Dlosses_impl.py256 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
311 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
360 logits.get_shape().assert_is_compatible_with(labels.get_shape())
428 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
495 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
567 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
650 predictions.get_shape().assert_is_compatible_with(labels.get_shape())
709 logits.get_shape().assert_is_compatible_with(multi_class_labels.get_shape())
776 logits.get_shape().assert_is_compatible_with(onehot_labels.get_shape())
822 labels_rank = labels.get_shape().ndims
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dadadelta_test.py79 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())
/external/tensorflow/tensorflow/python/kernel_tests/random/
Drandom_poisson_test.py118 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)
Drandom_gamma_test.py197 self.assertEqual([150], rnd.get_shape().as_list())
199 self.assertEqual([150, 2], rnd.get_shape().as_list())
201 self.assertEqual([150, 1, 2, 3], rnd.get_shape().as_list())
203 self.assertEqual([20, 30, 1, 2, 3], rnd.get_shape().as_list())
207 self.assertEqual([123, 2], rnd.get_shape().as_list())
212 self.assertEqual([None, 7, 3], rnd.get_shape().as_list())
216 self.assertEqual([None, None, None, 9, 6], rnd.get_shape().as_list())
221 self.assertIs(None, rnd.get_shape().ndims)
223 self.assertIs(None, rnd.get_shape().ndims)
/external/tensorflow/tensorflow/python/training/
Dinput_test.py778 self.assertAllEqual((1,), sparse.dense_shape.get_shape().as_list())
780 self.assertAllEqual((2,), batched.dense_shape.get_shape().as_list())
786 self.assertAllEqual((1,), sparse.dense_shape.get_shape().as_list())
788 self.assertAllEqual((1,), batched.dense_shape.get_shape().as_list())
796 self.assertIs(None, sparse.dense_shape.get_shape().num_elements())
798 self.assertIs(None, batched.dense_shape.get_shape().num_elements())
806 self.assertIs(None, sparse.dense_shape.get_shape().num_elements())
808 self.assertIs(None, batched.dense_shape.get_shape().num_elements())
813 self.assertAllEqual((8, 2), x["c"].get_shape().as_list())
911 self.assertAllEqual((1,), sparse.dense_shape.get_shape().as_list())
[all …]
Dadadelta_test.py89 self.assertEqual(slot[0].get_shape(), var0.get_shape())
93 self.assertEqual(slot_update[0].get_shape(), var0.get_shape())
97 self.assertEqual(slot[1].get_shape(), var1.get_shape())
101 self.assertEqual(slot_update[1].get_shape(), var1.get_shape())

12345678910>>...12