Home
last modified time | relevance | path

Searched refs:as_list (Results 1 – 25 of 296) 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 …]
Dcore_test.py78 self.assertListEqual([5, 2], outputs.get_shape().as_list())
102 self.assertListEqual([5, 4, 2], outputs.get_shape().as_list())
136 self.assertEqual(outputs.get_shape().as_list(), [5, 7])
140 self.assertEqual(outputs.get_shape().as_list(), [5, 2, 7])
144 self.assertEqual(outputs.get_shape().as_list(), [1, 2, 4, 7])
339 dense.compute_output_shape((None, 3)).as_list())
342 dense.compute_output_shape(ts([None, 3])).as_list())
345 dense.compute_output_shape(ts([None, 4, 3])).as_list())
460 self.assertEqual(y.get_shape().as_list(), [None, 6])
466 self.assertEqual(y.get_shape().as_list(), [1, 12])
[all …]
/external/tensorflow/tensorflow/python/keras/engine/
Dinput_spec_test.py40 [1, None, 2, 3], input_spec.to_tensor_shape(spec).as_list())
45 [None] * 5, input_spec.to_tensor_shape(spec).as_list())
49 [], input_spec.to_tensor_shape(spec).as_list())
53 [None, 3, 2], input_spec.to_tensor_shape(spec).as_list())
58 input_spec.to_tensor_shape(spec).as_list()
62 input_spec.to_tensor_shape(spec).as_list()
Ddeferred_sequential_test.py49 self.assertEqual(model.inputs[0].shape.as_list(), [2, 6])
50 self.assertEqual(model.outputs[0].shape.as_list(), [2, 2])
56 self.assertEqual(model.inputs[0].shape.as_list(), [None, 6])
57 self.assertEqual(model.outputs[0].shape.as_list(), [None, 2])
61 self.assertEqual(model.inputs[0].shape.as_list(), [None, 6])
62 self.assertEqual(model.outputs[0].shape.as_list(), [None, 2])
71 self.assertEqual(model.inputs[0].shape.as_list(), [None, 6])
72 self.assertEqual(model.outputs[0].shape.as_list(), [None, 2])
90 self.assertEqual(model.inputs[0].shape.as_list()[-1], 6)
91 self.assertEqual(model.outputs[0].shape.as_list()[-1], 2)
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_to_sparse_op_test.py123 self.assertEqual(st.indices.shape.as_list(), [7, 2])
124 self.assertEqual(st.values.shape.as_list(), [7])
125 self.assertEqual(st.dense_shape.shape.as_list(), [2])
130 self.assertEqual(st.indices.shape.as_list(), [4, 3])
131 self.assertEqual(st.values.shape.as_list(), [4])
132 self.assertEqual(st.dense_shape.shape.as_list(), [3])
136 self.assertEqual(st.indices.shape.as_list(), [7, 3])
137 self.assertEqual(st.values.shape.as_list(), [7])
138 self.assertEqual(st.dense_shape.shape.as_list(), [3])
Dragged_tensor_test.py109 self.assertEqual(rt.shape.as_list(), [2, None, 3])
188 self.assertEqual(rt.shape.as_list(), [5, None])
211 self.assertEqual(rt.shape.as_list(), [5, None])
213 self.assertEqual(rt.shape.as_list(), [None, None])
235 self.assertEqual(rt.shape.as_list(), [7, None])
256 self.assertEqual(rt.shape.as_list(), [5, None])
275 self.assertEqual(rt.shape.as_list(), [0, None])
277 self.assertEqual(rt.values.shape.as_list(), [0])
278 self.assertEqual(rt.value_rowids().shape.as_list(), [0])
288 self.assertEqual(rt.shape.as_list(), [5, None])
[all …]
/external/tensorflow/tensorflow/python/tpu/
Dtpu_sharding.py180 dims = shape.as_list()
187 (shape.as_list(), self._shard_dimension))
232 (shape.as_list(), self._shard_dimension))
233 dims = shape.as_list()
237 (shape.as_list(), self._shard_dimension))
240 (shape.as_list(), self._number_of_shards,
268 (shape.as_list(), self._shard_dimension))
269 dims = shape.as_list()
/external/tensorflow/tensorflow/python/ops/
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])
Dsobol_ops_test.py87 self.assertAllEqual([100, 10], s.shape.as_list())
95 assert s.shape.as_list() == [100, None]
106 assert s.shape.as_list() == [None, 10]
118 assert s.shape.as_list() == [None, None]
Dnn_grad_test.py73 inputs.get_shape().as_list(),
75 r_g.get_shape().as_list(),
85 x.get_shape().as_list(),
87 y.get_shape().as_list())
136 x.get_shape().as_list(),
138 y.get_shape().as_list())
/external/tensorflow/tensorflow/python/kernel_tests/
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 …]
Dreshape_op_test.py168 self.assertEqual([1, 1, None, 1], y.get_shape().as_list())
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())
212 self.assertEqual([50000**2], y.get_shape().as_list())
215 self.assertEqual([1, 50000**2], y.get_shape().as_list())
Dpad_op_test.py303 self.assertEqual([None, None, None, None], padded.get_shape().as_list())
308 self.assertEqual([None, None], padded.get_shape().as_list())
323 self.assertEqual([7, None], padded.get_shape().as_list())
328 self.assertEqual([None, 7], padded.get_shape().as_list())
333 self.assertEqual([7, None], padded.get_shape().as_list())
337 self.assertEqual([None, None], padded.get_shape().as_list())
343 self.assertEqual([None, None], padded.get_shape().as_list())
349 self.assertEqual([7, None], padded.get_shape().as_list())
355 self.assertEqual([6, None], padded.get_shape().as_list())
360 self.assertEqual([None, None, 20], padded.get_shape().as_list())
/external/tensorflow/tensorflow/python/kernel_tests/random/
Drandom_binomial_test.py98 self.assertEqual([0], rnd.shape.as_list())
104 self.assertEqual([10], rnd.shape.as_list())
106 self.assertEqual([], rnd.shape.as_list())
113 self.assertEqual([10], rnd.shape.as_list())
118 self.assertEqual([5, 2], rnd.shape.as_list())
125 self.assertEqual([10], rnd.shape.as_list())
132 self.assertEqual([10], rnd.shape.as_list())
139 self.assertEqual([10, 2, 3], rnd.shape.as_list())
146 self.assertEqual([10, 2, 3, 5], rnd.shape.as_list())
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())
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())
/external/tensorflow/tensorflow/python/keras/utils/
Dkernelized_utils_test.py50 shape = exact_kernel.shape.as_list()
64 shape = exact_kernel.shape.as_list()
78 shape = exact_kernel.shape.as_list()
95 shape = exact_kernel.shape.as_list()
108 shape = exact_kernel.shape.as_list()
/external/tensorflow/tensorflow/python/keras/layers/
Dmerge_test.py46 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
79 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
112 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
127 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
141 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
155 self.assertListEqual(o.shape.as_list(), [None, 4, 5])
170 self.assertListEqual(o.shape.as_list(), [None, 8, 5])
189 self.assertListEqual(unit_length_o.shape.as_list(), i1.shape.as_list())
205 self.assertListEqual(o.shape.as_list(), [None, 1])
221 self.assertListEqual(o.shape.as_list(), [None, 1])
[all …]
Dmulti_head_attention_test.py51 self.assertEqual(output.shape.as_list(), [None] + output_dims)
60 self.assertEqual(output.shape.as_list(), [None, 40, 80])
69 self.assertEqual(output.shape.as_list(), [None, 40, 80])
70 self.assertEqual(coef.shape.as_list(), [None, 12, 40, 40])
80 self.assertEqual(output.shape.as_list(), [None, 40, 80])
81 self.assertEqual(coef.shape.as_list(), [None, 12, 40, 60])
143 self.assertEqual(output.shape.as_list(), [None, 40, 80])
268 self.assertEqual(output.shape.as_list(), [None, 40, 80])
/external/tensorflow/tensorflow/python/keras/
Dbackend_test.py268 self.assertEqual(x.shape.as_list(), [3, 4])
270 self.assertEqual(x.shape.as_list(), [3, 4])
366 self.assertEqual(xy.shape.as_list(), [2, 4])
371 self.assertEqual(xy.shape.as_list(), [32, 28, 4])
387 self.assertEqual(tuple(xy.shape.as_list()), output_shape)
589 self.assertEqual(y.shape.as_list(), [1, 2, 5])
605 self.assertEqual(y.shape.as_list(), [1, 4, 4, 3])
610 self.assertEqual(y.shape.as_list(), [1, 3, 4, 4])
625 self.assertEqual(y.shape.as_list(), [1, 4, 4, 4, 3])
631 self.assertEqual(y.shape.as_list(), [1, 3, 4, 4, 4])
[all …]
/external/tensorflow/tensorflow/python/keras/integration_test/
Dlegacy_rnn_test.py75 self.assertEqual(outputs.shape.as_list(), [None, timestep, output_shape])
76 self.assertEqual(state.shape.as_list(), [None, output_shape])
108 self.assertEqual(outputs.shape.as_list(), [None, timestep, output_shape])
109 self.assertEqual(state.shape.as_list(), [None, output_shape])
141 self.assertEqual(outputs.shape.as_list(), [None, timestep, output_shape])
143 self.assertEqual(state[0].shape.as_list(), [None, output_shape])
144 self.assertEqual(state[1].shape.as_list(), [None, output_shape])
180 self.assertEqual(outputs.shape.as_list(), [None, timestep, output_shape])
184 self.assertEqual(state[0].shape.as_list(), [None, 2 * output_shape])
185 self.assertEqual(state[1].shape.as_list(), [None, 2 * output_shape])
[all …]
/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())
813 self.assertAllEqual((8, 2), x["c"].get_shape().as_list())
911 self.assertAllEqual((1,), sparse.dense_shape.get_shape().as_list())
913 self.assertAllEqual((2,), batched.dense_shape.get_shape().as_list())
919 self.assertAllEqual((1,), sparse.dense_shape.get_shape().as_list())
924 self.assertAllEqual((1,), batched.dense_shape.get_shape().as_list())
930 self.assertAllEqual((1,), sparse.dense_shape.get_shape().as_list())
[all …]
Dslot_creator_test.py47 self.assertEqual([2], slot.get_shape().as_list())
60 self.assertEqual([2], slot.get_shape().as_list())
75 self.assertEqual([2], slot.get_shape().as_list())
111 self.assertEqual([2], slot.get_shape().as_list())
155 self.assertEqual([2], slot.get_shape().as_list())
176 self.assertEqual([], slot.get_shape().as_list())
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_tensors_test.py173 w.shape.as_list(),
174 x.shape.as_list(),
175 y.shape.as_list(),
176 z.shape.as_list()
186 w.shape.as_list(),
187 x.shape.as_list(),
188 y.shape.as_list(),
189 z.shape.as_list()

12345678910>>...12