| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/ |
| D | fill_triangular_test.py | 63 self.assertAllEqual(y_.shape.as_list(), y_shape.as_list())
65 self.assertAllEqual(x_.shape.as_list(), x_shape.as_list())
68 self.assertAllEqual(y_shape_.as_list(), y_shape.as_list())
70 self.assertAllEqual(x_shape_.as_list(), x_shape.as_list())
73 b.forward_event_shape_tensor(x_shape.as_list()))
74 self.assertAllEqual(y_shape_tensor, y_shape.as_list())
76 b.inverse_event_shape_tensor(y_shape.as_list()))
77 self.assertAllEqual(x_shape_tensor, x_shape.as_list())
88 self.evaluate(b.forward_event_shape_tensor(x_shape_bad.as_list()))
94 self.evaluate(b.inverse_event_shape_tensor(y_shape_bad.as_list()))
|
| D | inline_test.py | 61 forward_event_shape_fn=lambda x: x.as_list() + [1],
70 y.as_list(),
71 bijector.forward_event_shape_tensor(x.as_list()).eval())
74 x.as_list(),
75 bijector.inverse_event_shape_tensor(y.as_list()).eval())
|
| /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/data/experimental/kernel_tests/ |
| D | rebatch_dataset_test.py | 48 [ts.as_list() for ts in _flat_shapes(dataset)])
51 [ts.as_list() for ts in _flat_shapes(rebatched_dataset)])
96 [ts.as_list() for ts in _flat_shapes(dataset)])
99 [ts.as_list() for ts in _flat_shapes(rebatched_dataset)])
110 [ts.as_list() for ts in _flat_shapes(dataset)])
113 [ts.as_list() for ts in _flat_shapes(rebatched_dataset)])
126 [ts.as_list() for ts in _flat_shapes(dataset)])
136 [ts.as_list() for ts in _flat_shapes(rebatched_dataset)])
150 [ts.as_list() for ts in _flat_shapes(dataset)])
153 [ts.as_list() for ts in _flat_shapes(rebatched_dataset)])
[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 …]
|
| D | inception_v2_test.py | 46 self.assertListEqual(logits.get_shape().as_list(),
49 self.assertListEqual(end_points['Predictions'].get_shape().as_list(),
59 self.assertListEqual(mixed_5c.get_shape().as_list(),
113 self.assertListEqual(end_points[endpoint_name].get_shape().as_list(),
143 original_depth = end_points[key].get_shape().as_list()[3]
144 new_depth = end_points_with_multiplier[key].get_shape().as_list()[3]
164 original_depth = end_points[key].get_shape().as_list()[3]
165 new_depth = end_points_with_multiplier[key].get_shape().as_list()[3]
187 self.assertListEqual(logits.get_shape().as_list(),
190 self.assertListEqual(pre_pool.get_shape().as_list(),
[all …]
|
| D | overfeat_test.py | 40 self.assertListEqual(logits.get_shape().as_list(),
51 self.assertListEqual(logits.get_shape().as_list(),
104 self.assertListEqual(logits.get_shape().as_list(),
107 self.assertListEqual(predictions.get_shape().as_list(), [batch_size])
119 self.assertListEqual(logits.get_shape().as_list(),
126 self.assertListEqual(logits.get_shape().as_list(),
130 self.assertEquals(predictions.get_shape().as_list(), [eval_batch_size])
|
| D | alexnet_test.py | 40 self.assertListEqual(logits.get_shape().as_list(),
51 self.assertListEqual(logits.get_shape().as_list(),
104 self.assertListEqual(logits.get_shape().as_list(),
107 self.assertListEqual(predictions.get_shape().as_list(), [batch_size])
119 self.assertListEqual(logits.get_shape().as_list(),
126 self.assertListEqual(logits.get_shape().as_list(),
130 self.assertEquals(predictions.get_shape().as_list(), [eval_batch_size])
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_to_sparse_op_test.py | 124 self.assertEqual(st.indices.shape.as_list(), [7, 2])
125 self.assertEqual(st.values.shape.as_list(), [7])
126 self.assertEqual(st.dense_shape.shape.as_list(), [2])
131 self.assertEqual(st.indices.shape.as_list(), [4, 3])
132 self.assertEqual(st.values.shape.as_list(), [4])
133 self.assertEqual(st.dense_shape.shape.as_list(), [3])
137 self.assertEqual(st.indices.shape.as_list(), [7, 3])
138 self.assertEqual(st.values.shape.as_list(), [7])
139 self.assertEqual(st.dense_shape.shape.as_list(), [3])
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| D | layers_test.py | 52 self.assertListEqual(masks[0].get_shape().as_list(),
57 self.assertListEqual(masked_weight[0].get_shape().as_list(),
78 self.assertListEqual(masks[ix].get_shape().as_list(), [
86 self.assertListEqual(masked_weight[ix].get_shape().as_list(), [
101 self.assertListEqual(masks[0].get_shape().as_list(),
106 self.assertListEqual(masked_weight[0].get_shape().as_list(),
126 self.assertListEqual(masks[ix].get_shape().as_list(), [
133 self.assertListEqual(masked_weight[ix].get_shape().as_list(), [
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| 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())
151 self.assertEqual([7, 2], stitched_t.get_shape().as_list())
235 self.assertEqual([2], stitched_t.get_shape().as_list())
254 self.assertEqual([7, 2], stitched_t.get_shape().as_list())
272 self.assertEqual([2], stitched_t.get_shape().as_list())
[all …]
|
| D | decode_raw_op_test.py | 37 self.assertEqual([2, None], decode.get_shape().as_list())
56 self.assertEqual([None, None], decode.get_shape().as_list())
85 self.assertEqual([None, None], decode.get_shape().as_list())
97 self.assertEqual([None, None], decode.get_shape().as_list())
110 self.assertEqual([None, None], decode.get_shape().as_list())
133 self.assertEqual([None, None], decode.get_shape().as_list())
|
| /external/tensorflow/tensorflow/python/ops/ |
| 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])
|
| D | nn_grad_test.py | 47 inputs.get_shape().as_list(),
49 r_g.get_shape().as_list(),
59 x.get_shape().as_list(),
61 y.get_shape().as_list())
110 x.get_shape().as_list(),
112 y.get_shape().as_list())
|
| D | math_grad_test.py | 91 value, shape, output, output.get_shape().as_list())
156 inputs, inputs.get_shape().as_list(),
157 outputs, outputs.get_shape().as_list())
167 inputs, inputs.get_shape().as_list(),
168 outputs, outputs.get_shape().as_list())
179 inputs, inputs.get_shape().as_list(),
180 outputs, outputs.get_shape().as_list())
191 inputs, inputs.get_shape().as_list(),
192 outputs, outputs.get_shape().as_list())
266 inputs.get_shape().as_list(), outputs,
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | merge_test.py | 41 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
74 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
107 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
122 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
136 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
150 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
165 self.assertListEqual(o.shape.as_list(), [None, 8, 5])
196 self.assertListEqual(o.shape.as_list(), [None, 1])
212 self.assertListEqual(o.shape.as_list(), [None, 1])
269 self.assertEqual(y.shape.as_list(), [None, 4, 5])
[all …]
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 282 self.assertEqual(xy.shape.as_list(), [2, 4])
287 self.assertEqual(xy.shape.as_list(), [32, 28, 4])
293 self.assertEqual(xy.shape.as_list(), [32, 1, 30])
451 self.assertEqual(y.shape.as_list(), [1, 2, 5])
469 self.assertEqual(y.shape.as_list(), [1, 4, 4, 3])
477 self.assertEqual(y.shape.as_list(), [1, 3, 4, 4])
497 self.assertEqual(y.shape.as_list(), [1, 4, 4, 4, 3])
506 self.assertEqual(y.shape.as_list(), [1, 3, 4, 4, 4])
519 self.assertEqual(y.shape.as_list(), [1, 9, 2])
525 self.assertEqual(y.shape.as_list(), [2, None, 2])
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_sharding.py | 178 (shape.as_list(), self._shard_dimension))
179 dims = shape.as_list()
183 (shape.as_list(), self._shard_dimension))
186 (shape.as_list(), self._number_of_shards,
214 (shape.as_list(), self._shard_dimension))
215 dims = shape.as_list()
|
| /external/tensorflow/tensorflow/contrib/specs/python/ |
| D | specs_test.py | 45 self.assertEqual(outputs.get_shape().as_list(), [1, 18, 19, 64])
60 self.assertEqual(outputs.get_shape().as_list(), [17, 55])
70 self.assertEqual(outputs.get_shape().as_list(), [17, 10])
84 self.assertEqual(outputs.get_shape().as_list(), [1, 8, 8, 5])
97 self.assertEqual(outputs.get_shape().as_list(), [1, 8, 8, 5])
114 self.assertEqual(outputs.get_shape().as_list(), [1, 8, 8, 5])
130 self.assertEqual(outputs.get_shape().as_list(), [10, 30])
146 self.assertEqual(outputs.get_shape().as_list(), [10, 20])
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| D | graph_io_test.py | 223 self.assertAllEqual((batch_size,) + shape, feature.get_shape().as_list())
254 self.assertAllEqual((None,), inputs.get_shape().as_list())
287 self.assertAllEqual((batch_size,), inputs.get_shape().as_list())
337 self.assertAllEqual((None,), inputs.get_shape().as_list())
389 self.assertAllEqual((None,), keys.get_shape().as_list())
391 self.assertAllEqual((None,), result["sequence"].get_shape().as_list())
446 self.assertAllEqual((None,), result["sequence"].get_shape().as_list())
487 self.assertAllEqual((None,), inputs.get_shape().as_list())
531 self.assertAllEqual((None,), keys.get_shape().as_list())
532 self.assertAllEqual((None,), inputs.get_shape().as_list())
[all …]
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | kernelized_utils_test.py | 50 shape = exact_kernel.get_shape().as_list()
64 shape = exact_kernel.get_shape().as_list()
78 shape = exact_kernel.get_shape().as_list()
95 shape = exact_kernel.get_shape().as_list()
108 shape = exact_kernel.get_shape().as_list()
|
| /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())
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers_test.py | 70 self.assertListEqual(output.get_shape().as_list(), [5, 1, 2, 3])
76 self.assertListEqual(output.get_shape().as_list(), [5, 2, 1, 2])
91 self.assertListEqual(output.get_shape().as_list(), [5, 1, 2, 3])
103 self.assertListEqual(output.get_shape().as_list(), [5, 2, 3, 3])
110 self.assertListEqual(output.get_shape().as_list(), [5, 3, 2, 3])
116 self.assertListEqual(output.get_shape().as_list(), [5, height, width, 3])
122 self.assertListEqual(output.get_shape().as_list(), [5, 1, 1, 3])
139 self.assertListEqual(output.get_shape().as_list(), [5, 1, 2, 4, 3])
146 self.assertListEqual(output.get_shape().as_list(), [5, 2, 1, 2, 4])
162 self.assertListEqual(output.get_shape().as_list(), [5, 1, 2, 4, 3])
[all …]
|