/external/tensorflow/tensorflow/python/keras/legacy_tf_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 | 76 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 …]
|
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) 558 x.get_shape().as_list(), [all …]
|
D | reshape_op_test.py | 40 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 …]
|
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 | 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()) 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 …]
|
D | constant_op_eager_test.py | 191 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 …]
|
D | constant_op_test.py | 199 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/ |
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()) 303 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]) 209 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 …]
|
D | exponential_test.py | 57 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/ |
D | embedding_ops.py | 69 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 …]
|
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 | 256 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/ |
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())
|
/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)
|
D | random_gamma_test.py | 197 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/ |
D | input_test.py | 778 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 …]
|
D | adadelta_test.py | 89 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())
|