| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional_test.py | 40 def _run_test(self, kwargs, expected_output_shape): argument
50 expected_output_shape=expected_output_shape)
52 def _run_test_extra_batch_dim(self, kwargs, expected_output_shape): argument
58 if expected_output_shape is not None:
59 expected_output_shape = (None,) + expected_output_shape
65 expected_output_shape=expected_output_shape)
98 def test_conv1d(self, kwargs, expected_output_shape, requires_gpu=False): argument
102 self._run_test(kwargs, expected_output_shape)
103 self._run_test_extra_batch_dim(kwargs, expected_output_shape)
177 def _run_test(self, kwargs, expected_output_shape, spatial_shape=(7, 6)): argument
[all …]
|
| D | einsum_dense_test.py | 229 expected_output_shape): argument
230 del expected_output_shape # Not used in this test.
240 expected_output_shape): argument
254 self.assertAllEqual(expected_output_shape, output_tensor.shape.as_list())
|
| D | recurrent_test.py | 861 expected_output_shape = [(None, timesteps, 6),
868 expected_output_shape)
876 expected_output_shape = [(None, timesteps, 6),
883 expected_output_shape)
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | category_encoding_test.py | 92 expected_output_shape = [None, num_tokens]
99 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
116 expected_output_shape = [None, num_tokens]
124 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
203 expected_output_shape = [None, num_tokens]
211 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
267 expected_output_shape = [None, num_tokens]
271 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
281 expected_output_shape = [None, num_tokens]
285 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
[all …]
|
| D | discretization_test.py | 55 expected_output_shape = [None, 4]
60 self.assertAllEqual(expected_output_shape, bucket_data.shape.as_list())
70 expected_output_shape = [None, 4]
75 self.assertAllEqual(expected_output_shape, bucket_data.shape.as_list())
100 expected_output_shape = [None, None]
105 self.assertAllEqual(expected_output_shape, bucket_data.shape.as_list())
116 expected_output_shape = [None, None]
121 self.assertAllEqual(expected_output_shape, bucket_data.shape.as_list())
|
| D | text_vectorization_test.py | 837 expected_output_shape = [None, None]
848 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
864 expected_output_shape = [None, output_sequence_length]
876 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
891 expected_output_shape = [None, output_sequence_length]
903 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
918 expected_output_shape = [None, output_sequence_length]
930 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
954 expected_output_shape = [None, max_tokens]
965 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
[all …]
|
| D | discretization_distribution_test.py | 46 expected_output_shape = [None, 4]
54 self.assertAllEqual(expected_output_shape, bucket_data.shape.as_list())
|
| D | image_preprocessing_test.py | 56 expected_output_shape=(None, expected_height, expected_width,
176 expected_output_shape=(None, expected_height, expected_width,
230 expected_output_shape=(None, expected_height, expected_width,
320 expected_output_shape=(None, 5, 6, 3))
522 expected_output_shape=(None, orig_height, orig_width, channels))
1070 expected_output_shape=(None, orig_height, orig_width, channels))
1123 expected_output_shape=(None, orig_height, orig_width, channels))
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 99 expected_output_shape=None, argument
219 if expected_output_shape is not None:
220 assert_shapes_equal(tensor_shape.TensorShape(expected_output_shape),
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | functional_test.py | 1517 def _testShapeInference(self, model, input_shape, expected_output_shape): argument
1520 self.assertEqual(output_value.shape, expected_output_shape)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 5270 std::vector<int> expected_output_shape; in TEST_P() member
5455 TestOpConverter("my_gather", node_def, p.expected_output_shape, p.status, in TEST_P()
|