| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | categorical_encoding_test.py | 58 expected_output_shape = [None, max_tokens]
64 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
78 expected_output_shape = [None, max_tokens]
85 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
99 expected_output_shape = [None, max_tokens]
105 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
119 expected_output_shape = [None, max_tokens]
126 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
143 expected_output_shape = [None, max_tokens]
150 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
[all …]
|
| D | text_vectorization_test.py | 569 expected_output_shape = [None, None]
580 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
596 expected_output_shape = [None, output_sequence_length]
608 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
623 expected_output_shape = [None, output_sequence_length]
635 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
650 expected_output_shape = [None, output_sequence_length]
662 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
686 expected_output_shape = [None, max_tokens]
697 self.assertAllEqual(expected_output_shape, int_data.shape.as_list())
[all …]
|
| D | image_preprocessing_test.py | 52 expected_output_shape=(None, expected_height, expected_width,
125 expected_output_shape=(None, expected_height, expected_width,
180 expected_output_shape=(None, expected_height, expected_width,
266 expected_output_shape=(None, 5, 6, 3))
478 expected_output_shape=(None, orig_height, orig_width, channels))
531 expected_output_shape=(None, orig_height, orig_width, channels))
572 expected_output_shape=(None, orig_height, orig_width, channels))
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional_test.py | 35 def _run_test(self, kwargs, expected_output_shape): argument
45 expected_output_shape=expected_output_shape)
58 def test_conv1d(self, kwargs, expected_output_shape): argument
61 self._run_test(kwargs, expected_output_shape)
102 def _run_test(self, kwargs, expected_output_shape): argument
113 expected_output_shape=expected_output_shape)
126 def test_conv2d(self, kwargs, expected_output_shape=None): argument
130 self._run_test(kwargs, expected_output_shape)
171 def _run_test(self, kwargs, expected_output_shape): argument
183 expected_output_shape=expected_output_shape)
[all …]
|
| D | recurrent_test.py | 857 expected_output_shape = [(None, timesteps, 6),
864 expected_output_shape)
872 expected_output_shape = [(None, timesteps, 6),
879 expected_output_shape)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | testing_utils.py | 77 expected_output_dtype=None, expected_output_shape=None, argument
172 if expected_output_shape is not None:
173 assert_shapes_equal(tensor_shape.TensorShape(expected_output_shape),
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 4584 std::vector<int> expected_output_shape; in TestConvertGather() member
4720 const auto& expected_output_shape = ok_params[i].expected_output_shape; in TestConvertGather() local
4723 TrtWeightDimsNumElements(GetTestDims(expected_output_shape))); in TestConvertGather()
4725 expected_output_shape.begin() + 1, expected_output_shape.end()); in TestConvertGather()
4745 /*batch_size=*/expected_output_shape[0]); in TestConvertGather()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | network_test.py | 1175 def _testShapeInference(self, model, input_shape, expected_output_shape): argument
1178 self.assertEqual(output_value.shape, expected_output_shape)
|